重建二叉树

《剑指offer》面试题7：重建二叉树

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

思路：前序遍历序列的第一个数为二叉树根节点，在中序遍历序列中，根节点左边的序列为左子树中序遍历序列，根节点右边的序列为右子树中序遍历序列。例如1为根节点，在中序遍历序列中，从左到右依次遍历序列，直到找到根节点，根节点1左边的{4,7,2}为左子树，右边的{5,3,8,6}为右子树，其先序序列分别为{2,4,7}，{3,5,6,8}。同样的方法，使用递归构建左右子树的左右子树。

代码如下：

public TreeNode reConstructBinaryTree(int[] pre,int[] in){
    if (pre == null || in == null || pre.length != in.length || pre.length < 1) {
        return null;
    }
    return reConstruct(pre,0,pre.length - 1,in,0,in.length - 1);
}

public TreeNode reConstruct(int[] pre,int preStart,int preEnd,int[] in,int inStart,int inEnd) {
    if (preStart > preEnd || inStart > inEnd) {
        return null;
    }
    TreeNode root = new TreeNode(pre[preStart]);
    for (int i = inStart;i < inEnd;i++) {
        if (in[i] == pre[preStart]) {
            root.left = reConstruct(pre,preStart + 1,i - inStart + preStart,in,inStart,i - 1);
            root.right = reConstruct(pre,i - inStart + preStart + 1,preEnd,in,i + 1,inEnd);
            break;
        }
    }
    return root;
}