#105从前序与中序遍历序列构造二叉树

import java.util.*;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int val) {
        this.val = val;
    }
}

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder.length != inorder.length) {
            throw new IllegalArgumentException("Traversal lengths differ");
        }
        Map<Integer, Integer> index = new HashMap<>();
        for (int i = 0; i < inorder.length; i++) index.put(inorder[i], i);
        return build(preorder, 0, 0, inorder.length - 1, index);
    }

    private TreeNode build(int[] preorder, int preStart, int inLeft,
                           int inRight, Map<Integer, Integer> index) {
        if (inLeft > inRight) return null;
        int rootValue = preorder[preStart];
        int pivot = index.get(rootValue);
        int leftSize = pivot - inLeft;
        TreeNode root = new TreeNode(rootValue);
        root.left = build(preorder, preStart + 1, inLeft, pivot - 1, index);
        root.right = build(preorder, preStart + leftSize + 1,
                pivot + 1, inRight, index);
        return root;
    }
}