重建二叉树

题目描述

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

时间限制：C/C++ 1秒，其他语言2秒

解题思路：前序遍历的序列中，第一个数字必然是树的根节点，而该根节点在中序遍历的序列中其左边为左子树，右边为右子树。重建树的过程中，要注意数组的边界。

解法1：不使用内置函数

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
       return  ConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
    }
    public static TreeNode ConstructBinaryTree(int [] pre,int starpre,int endpre,int [] in,int starin,int endin)
    {
      if(starpre>endpre||starin>endin) return null;
      TreeNode root=new TreeNode(pre[starpre]);
    
      for(int i=starin;i<=endin;i++)  //从中序序列中找出根节点的下标
      {
        if(in[i]==pre[starpre])
        {
         
          root.left=ConstructBinaryTree(pre,starpre+1,i+starpre-starin,in,starin,i-1); //注意前序数组的下标，可以自己在草稿纸上试一下，就知道怎么来的了
          root.right=ConstructBinaryTree(pre,i+starpre-starin+1,endpre,in,i+1,endin);     
         
        }
        
      }
      return root;
    }

}

解法2：使用内置函数：注意：copyOfRange函数是左闭右开分割数组的

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
       if(pre.length == 0||in.length == 0){
            return null;
        }
        TreeNode node = new TreeNode(pre[0]);
        for(int i = 0; i < in.length; i++){
            if(pre[0] == in[i]){
                node.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i+1), Arrays.copyOfRange(in, 0, i));
                node.right = reConstructBinaryTree(Arrays.copyOfRange(pre, i+1, pre.length), Arrays.copyOfRange(in, i+1,in.length));
            }
        }
        return node;
    }
}