Leetcode105. 从前序与中序遍历序列构造二叉树

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

Leetcode105

题意：

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

分析

1. 关于树的构建或者遍历通常做法都是递归
2. 通过循环分析前序遍历和中序遍历可构建二叉树
   
3. 分析要达到上述循环构建二叉树我们需要什么参数。
   （1）首先需要知道root.index 通过root.index在前序数组中找到值构建节点 在通过节点值找到其在中序遍历中的midindex，则中序遍历中的midindex左边为左子树，右边为右子树
   （2） 因为需要其左子树的边界值和右子树的边界值，所以可根据中序遍历中midindex到其左边界的个数来知晓左子树的个数，从而直接前序遍历左子树中右边界的值
4. 进一步抽象出来我们所需要的参数，首先是前序遍历数组preorder和中序遍历数组inorder，以及前序遍历左下标preLeft、前序遍历右下标preRight、中序遍历左下标iLeft和中序遍历右下标iRight。
   
5. 简单演算可知前序遍历左子树中右边界的值为midIndex-iLeft+preLeft

代码

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder.length == 0 || inorder.length == 0 || preorder.length != inorder.length){
            return null;
        }
        //因为我们整个过程只需要用到中序遍历的下标 所以可以用map存储中序遍历的值和下标 可减少一定的时间复杂度
        HashMap<Integer,Integer> map = new HashMap<>();
     
        for(int i = 0;i<inorder.length;i++){
            map.put(inorder[i],i);
        }
        return helper(preorder,0,preorder.length-1,map,0,inorder.length-1);
        
    }
    public TreeNode helper(int[] preorder,int preLeft,int preRight,HashMap<Integer,Integer> map,int iLeft,int iRight){
        //递归结束条件
        if(preLeft>preRight){
            return null;
        }
        int mid = preorder[preLeft];
        TreeNode root = new TreeNode(mid);
        int midIndex = map.get(mid);
        root.left = helper(preorder,preLeft+1,midIndex-iLeft+preLeft,map,iLeft,midIndex-1);
        root.right = helper(preorder,midIndex-iLeft+preLeft+1,preRight,map,midIndex+1,iRight);
        return root;
    }
}

执行结果