【剑指OFFER】重建二叉树

【题目描述】

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

时间限制：1秒 空间限制：32768K

 

【AC代码】

Reference: https://www.nowcoder.com/questionTerminal/8a19cbe657394eeaac2f6ea9b0f6fcf6?f=discussion

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode root = reConstructBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
        return root;
    }
    private TreeNode reConstructBinaryTree(int [] pre, int startPre, int endPre, int [] in, int startIn, int endIn) {
        if (startPre > endPre || startIn > endIn) return null; //验证子树是否整理完毕
        TreeNode root = new TreeNode(pre[startPre]); //每颗子树的根节点肯定是pre子数组的首元素，所以每次新建一个子树的根节点。
        //根据中序遍历找到左子树和右子树的长度
        for (int i = startIn; i <= endIn; i++) {
            if (in[i] == root.val) {
                //利用递归思想，每次将左右子树当成新二叉树进行处理。
                //(i-startIn)指中序根节点i与前序起点startIn之间的左孩子长度
                root.left = reConstructBinaryTree(pre, startPre+1, startPre+(i-startIn), in, startIn, i-1);
                root.right = reConstructBinaryTree(pre, startPre+(i-startIn)+1, endPre, in, i+1, endIn);
                break;
            }
        }
        return root;
    }
}