剑指offer JZ4重建二叉树

重建二叉树

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

输入
[1,2,3,4,5,6,7],[3,2,4,1,6,5,7]
返回值
{1,2,5,3,4,6,7}

思路

根据前序数组第一个值，寻找根节点的位置。
根据根节点的位置将前序和中序序列拆分成 左右子树的前、中序列
利用递归，通过左子树的前、中序列构建左子树，通过右子树的前、中序列构建右子树

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        //根据前序遍历数组的第一个值，将中序遍历的数组进行分割，先构建左子树，再构建右子树
        if(pre.length==0){
            return null;
        }
        TreeNode node = new TreeNode(pre[0]) ;
        if(pre.length==1){
            node.left = null;
            node.right = null;
            return node;
        }
        
        int root = 0;
        //从中序遍历中找出根的位置
        for(int i=0;i<in.length;i++){
            if(in[i]==pre[0]){
                root = i;
            }
        }
        
        //提取左子树前序、中序遍历，右子树前序、中序遍历
        int in_left[] = new int[root];
        int pre_left[] = new int[root];
        int in_right[] = new int[in.length - root -1];
        int pre_right[] = new int[in.length - root -1];
        for(int i=0;i<in.length;i++){
            if(i<root){    // 中序序列 左边的序列
                in_left[i] = in[i];
            }
            if(i>root){    // 中序序列 右边的序列
                in_right[i-root-1] = in[i];
            }
        }
        for(int i=1;i<pre.length;i++){ 
            if(i<=root){            //前序序列左
                pre_left[i-1] = pre[i];
            }
            if(i>root){            //前序序列右
                pre_right[i-root-1] = pre[i];
            }
        }
        
        node.left = reConstructBinaryTree(pre_left,in_left);
        node.right = reConstructBinaryTree(pre_right,in_right);
        
        return node;
    }
}