leetcode 根据前序遍历和中序遍历重建二叉树

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //画出图，找出root的索引，然后对应找出前序和中序关于索引的边界
    // 存储 inorder 中元素和索引
    Map<Integer,Integer> map = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for(int i = 0;i < inorder.length;i++){
            map.put(inorder[i],i);
        }
        return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
    }
    /**
    定义：前序遍历的数组为：preorder[preStart...preEnd]
         中序遍历的数组为: inorder[inStart...inEnd]
         构造这个二叉树并返回该二叉树的根节点
     */
    TreeNode build(int[] preorder,int preStart,int preEnd,int[] inorder,int inStart,int inEnd){
        if(preStart>preEnd){
            return null;
        }
        //root节点对应的值就是前序遍历数组的第一个元素
        int rootValue = preorder[preStart];
        //rootVal在中序遍历数组中的索引
        int index = map.get(rootValue);

        int leftSize = index - inStart; //左边元素的个数
        
        // 先构造出当前根节点
        TreeNode root = new TreeNode(rootValue);

        //preorder,preStart+1,preStart+leftSize, 前序中的左子树
        //inorder,inStart，index-1 ，中序中的左子树
        root.left = build(preorder,preStart+1,preStart+leftSize,inorder,inStart,index-1);

        //preorder,preStart+leftSize+1,preEnd,前序中的右子树
        //inorder,index+1,inEnd,中序中的右子树
        root.right = build(preorder,preStart+leftSize+1,preEnd,inorder,index+1,inEnd);

        return root;
    }
}