JAVA-牛客-重建二叉树

重建二叉树

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

分析

通过前序遍历和中序遍历去重建二叉树
前序遍历代表根在最前方，中序遍历根在其左子树和右子树的中间。
我们通过创建根节点的方式去重建整个二叉树，
要用根来创建二叉树，需要对前序进行遍历，整个问题可以化分为数个子问题进行解决。
创建一个二叉树，首先创建根节点，其次创建左子树和右子树。
从上面数据，首先找到根节点->1，其在中序遍历的位置4，7，2,->1->5,3,8,6.
其左子树为4，7，2.右子树为5，3，8，6.
此时，由先序遍历可知，1的左子树为2，4，7.由此可以得出，2为1的左子根，
而1的右子根从前序遍历中在左子树的节点之后，右子根为3.
继续划分子问题，将先序遍历中1的左右子树也找出来，
左子树2，4，7，右子树为3，5，6，8
先将左子树按照上述方式进行创建，
2划分左右子树，4，7为2的左子树，在中序遍历中2的右边为1，因此没有右子树。
在先序遍历中2之后为4，且在中序遍历中4在2的左边，因此4为2的左子树，
又有7为4的子树，7在4的右边，7为4的右子树。
以此推导，按照根在先序遍历中的顺序来创建节点，按照中序遍历中的顺序来判断其是左子树还是右子树。

代码如下：

import java.util.Arrays;
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]);
        buildTree(root,pre,in);
        return root;
        
    }
    public void buildTree(TreeNode root,int[] pre,int[] in){
        
        int i =0;
        for(;i<in.length;i++){
            if(in[i]==root.val){
                break;
            }
        }
        if(i>0){
            root.left = new TreeNode(pre[1]);
            buildTree(root.left,Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
        }
        if(in.length-i-1>0){
            root.right = new TreeNode(pre[i+1]);
            buildTree(root.right,Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
        }

        
        
    }
}