二叉树系列之「重建二叉树」

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列 {1,2,4,7,3,5,6,8} 和中序遍历序列 {4,7,2,1,5,3,8,6} ，则重建二叉树并返回。

//Definition for binary tree
public class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int val) { 
        this.val = val; 
     }
 }

可以先看解法 ①，了解思路之后，再看解法②，最终需要掌握的也是解法②

解法 ①
比较好理解，但是空间复杂度为 O(n)，因为用到了 Arrays.copyOfRange()

public class Solution {
  
    public TreeNode reConstructBinaryTree(int[] pre, int[] in ) {
      
        if(pre.length == 0 || in.length == 0) return null;
      
        TreeNode root = new TreeNode(pre[0]);
      
        for(int i=0; i<in.length; i++){ 
            if(in[i] == pre[0]){  // 注意：数组 copy 是包前不包后
                root.left = reConstructBinaryTree(
                  Arrays.copyOfRange(pre,1,i+1),
                  Arrays.copyOfRange(in,0,i));
                root.right = reConstructBinaryTree(
                  Arrays.copyOfRange(pre,i+1,pre.length),
                  Arrays.copyOfRange(in,i+1,in.length));
              
                break;
            }
        }
        return root;
    }
}

解法 ②
解法 ② 的框架跟解法 ① 差不多，只不过需要处理一些细节问题
空间复杂度 O(1)

public class Solution {
    public TreeNode reConstructBinaryTree(int[] pre,int[] in ) {
        //if(pre.length == 0 || in.length == 0) return null;
        return reBuild (pre, 0, pre.length-1, in, 0, in.length-1);
    }
  
    public static TreeNode reBuild (int[] pre,int pre_left,int pre_right,
                            int[] in,int in_left,int in_right){
        
     	  if(pre_left > pre_right) return null;
        TreeNode root = new TreeNode(pre[pre_left]); // 找到根节点
        for(int i=in_left; i<in_right; i++){
            if(in[i] == root.val){
                root.left = reBuild(pre,pre_left+1,pre_left+i-in_left, 
                                    in,in_left,i-1);
                root.right = reBuild(pre,pre_left+i-in_left+1,pre_right, 
                                     in,i+1,in_right);
                break;
            }
        }
        return root;
    }
}

总结：

• 仔细观察不难发现此题类型有点类似快排
• 共同点是都是递归实现，且涉及到数组区间的变化
• 当遇到这种情况时，可以用左右指针 left 和 right 代替数组一定区间