JZ7 重建二叉树

描述

给定节点数为 n 的二叉树的前序遍历和中序遍历结果，请重建出该二叉树并返回它的头结点。

例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建出如下图所示。

提示:

1.vin.length == pre.length

2.pre 和 vin 均无重复元素

3.vin出现的元素均出现在 pre里

4.只需要返回根结点，系统会自动输出整颗树做答案对比

数据范围： n≤2000，节点的值  −10000≤val≤10000

要求：空间复杂度 O(n) ，时间复杂度 O(n) 

示例1

输入：

[1,2,4,7,3,5,6,8],[4,7,2,1,5,3,8,6]

复制返回值：

{1,2,3,4,#,5,6,#,7,#,#,8}

复制说明：

返回根节点，系统会输出整颗二叉树对比结果，重建结果如题面图示    

示例2

输入：

[1],[1]

返回值：

{1}

示例3

输入：

[1,2,3,4,5,6,7],[3,2,4,1,6,5,7]

返回值：

{1,2,5,3,4,6,7}

经典 中序+x序=》还原二叉树问题 c++敲过好多遍了，用java敲一次，可能有失误，但是算法还记得，也能独立敲出来，就是需要2、3次的Dbug

我的代码:

 public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
        if(pre.length==0) return null;//健壮性 数组为空直接返回null
        TreeNode root=new TreeNode(pre[0]);//先序第一个是根
        //拆分出新的左右子树 pre和vin数组 递归创建左右子树
        int k=0;
        while (vin[k]!=pre[0]) k++;
        if(k>0){//如果还有左子树   左子树长度正好为k
            int[] preL=Arrays.copyOfRange(pre,1,k+1);//左边长度正好为k [1,k+1)
            int[] vinL= Arrays.copyOfRange(vin,0,k);//[0,k)
            root.left=reConstructBinaryTree(preL,vinL);//递归创建左右子树
        }
        if(pre.length-k-1>0){//如果还有右子树 注意:右子树长度 len-k-1 还要减一 根已经占了一个
            int[] preR=Arrays.copyOfRange(pre,k+1,pre.length);//剩下的直接右子树的了  注意也是k+1开始
            int[] vinR=Arrays.copyOfRange(vin,k+1,vin.length);//[k+1,length)
            root.right=reConstructBinaryTree(preR,vinR);
        }

        return root;//返回根
    }

语法层面简化，使得看起来很厉害的亚子:

    public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
        if(pre.length==0) return null;//健壮性 数组为空直接返回null
        TreeNode root=new TreeNode(pre[0]);//先序第一个是根
        int k=0;while (vin[k]!=pre[0]) k++;
        if(k>0) root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,k+1),Arrays.copyOfRange(vin,0,k));//递归创建左右子树
        if(pre.length-k-1>0) root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,k+1,pre.length),Arrays.copyOfRange(vin,k+1,vin.length));
        return root;//返回根
    }