根据前序和中序遍历序列重建二叉树

题目描述

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

解题思路：

采用递归调用的思想，分别递归调用重建左右子树。

第一次递归：

左子树的前序为：{2，4，7}，中序为{4,7,2}，根据左子树的前序和中序递归调用方法重建左子树；

右子树的前序为：{3，5，6,8}，中序为{5,3,8,6}，根据右子树的前序和中序递归调用方法重建右子树。

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class Solution {
    
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode tree=new TreeNode(pre[0]);
        int count=0;
        for(int i=0;i<in.length;i++){
            if(in[i]==pre[0]){
                break;
            }
            count++;
        }
        //获取左子树的先序
        int[] lpre=new int[count];
        for(int i=0;i<count;i++){
            lpre[i]=pre[i+1];
        }
        //获取左子树的中序
        int [] lin=new int[count];
        for(int i=0;i<count;i++){
            lin[i]=in[i];
        }
        //递归调用重建左子树
        if(count>0){
            tree.left=reConstructBinaryTree(lpre,lin);
        }
        
        //获取右子树的先序
        int[] rpre=new int[pre.length-count-1];
        for(int i=0;i<rpre.length;i++){
            rpre[i]=pre[i+count+1];
        }
        //获取右子树的中序
        int [] rin=new int[pre.length-count-1];
        for(int i=0;i<pre.length-count-1;i++){
            rin[i]=in[i+count+1];
        }
        //递归调用重建左子树
        if(pre.length-count-1>0){
            tree.right=reConstructBinaryTree(rpre,rin);
        }
        return tree;
    }
}