剑指Offer（书）：重建二叉树

Posted 2020-12-28 liter7

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

分析：递归的本质就是将大问题不断分解为相同类型的小问题，直到找到一个出口为止。这里，根据前序和中序的性质找到1为根节点，那么就可以分解为2，4，7与4，7，2，即长度变短了。之后依次求解其他的。

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

    TreeNode treeNode = new TreeNode(pre[0]);
    for (int i = 0; i < in.length; i++) {
        if (pre[0] == in[i]) {
            treeNode.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(in, 0, i));
            treeNode.right = reConstructBinaryTree(Arrays.copyOfRange(pre, i + 1, pre.length), Arrays.copyOfRange(in, i + 1, in.length));
        }
    }
    return treeNode;
}