【LeetCode】重建二叉树day04

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

 

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / 
  9  20
    /  
   15   7
 

限制：

0 <= 节点个数 <= 5000

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
    //如果前序遍历数组为空那这棵树也是空的
        if(preorder == null ){
            return null ;
        }else{
        //先拿到中序便利的数量
            int length = inorder.length;
        //通过Map记录中序遍历的值，在后面可以根据根节点拿到左子树和又字树的元素
         Map<Integer,Integer> inorderMap = new HashMap<Integer,Integer>();
            for(int i = 0 ;i<length;i++){
                inorderMap.put(inorder[i],i);
            }
            // 构建一个树根据前序遍历和后续遍历的数组
            //其实有人会问为什么要这样写这么多参数。因为我们要递归调用将其中的遍历的数组切开，然后去便利左子树右子树。
          return  buildTree(preorder,0,length-1,inorder,0,length-1,inorderMap);
        }
    }
      public  TreeNode buildTree(int [] preorder,int preorderStart,int preorderEnd,int[] inorder,int inoderStart, int inorderEnd,Map<Integer,Integer> inorderMap){
        if(preorderStart>preorderEnd){
            return null;
        }
        // 拿到根节点
        int rootVal  = preorder[preorderStart];
        // 创建一颗有根节点的二叉树
        TreeNode root = new TreeNode(rootVal);
        if(preorderStart == preorderEnd){
            return root;
        }else{
        拿到根节点在遍历数组中的位置
            int rootIndex =  inorderMap.get(rootVal);
            // 左子树的节点数量
           int leftNodes =  rootIndex - inoderStart; 右子树节点的数量
           int  rightNodes = inorderEnd - rootIndex;
            构建左子树
            TreeNode leftSubtree = buildTree(preorder, preorderStart + 1, preorderStart + leftNodes, inorder, inoderStart, rootIndex - 1, inorderMap);
            便利右子树
            TreeNode rightSubtree = buildTree(preorder, preorderEnd - rightNodes + 1, preorderEnd, inorder, rootIndex + 1, inorderEnd, inorderMap);
            //将左子树与右子树拼起来
            root.left = leftSubtree;
            root.right = rightSubtree;
        }
        return  root;

    }



}