剑指Offer(书):重建二叉树

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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;
}

猜你喜欢

转载自www.cnblogs.com/liter7/p/9420294.html