重建二叉树

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

前序遍历：中左右
中序编列：左中右
后续编列：左右中
->点我

思路

在前序排序遍历序列中,第一个总是数的根节点,但在中序编序列中,根节点的值在序列的中间,
在中序编序列中,左子树的节点的值位于根节点值的左边,右子树的节点的值位于根节点的右边

实现

public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
    return reBuild(pre, in, 0, pre.length - 1, 0, in.length - 1);
}
private TreeNode reBuild(int[] pre, int[] in, int preHead, int preEnd, int inHead, int inEnd) {
    TreeNode node = new TreeNode(pre[preHead]);
    if (preHead > preEnd || inHead > inEnd) {//出口
        return null;
    }

    int index = inHead;
    while (node.val != in[index] && index <= inEnd) {
        index++;
    }
    if (index > inEnd) {
        throw new RuntimeException("数组不符合规范！");
    }
    int leftLength = index - inHead;
    if (leftLength > 0) {//有 左子树
        node.left = reBuild(pre, in, preHead + 1, preHead + leftLength,
                inHead, index - 1);
    }
    if (leftLength < inEnd - inHead) {//有 右子树
        node.right = reBuild(pre, in, preHead + leftLength + 1, preEnd, index + 1, inEnd);
    }
    return node;
}

测试用例

public void test() {
    int[] pre = { 1, 2, 4, 7, 3, 5, 6, 8 };
    int[] in = { 4, 7, 2, 1, 5, 3, 8, 6 };
    TreeNode root = reConstructBinaryTree(pre, in);
    System.out.print("test:");
    preOrderTraverse(root);
    System.out.print("//");
    inOrderTraverse(root);
    System.out.println();
}
private void preOrderTraverse(TreeNode node) {
    if (node == null)
        return;
    System.out.print(node.val);
    preOrderTraverse(node.left);
    preOrderTraverse(node.right);
}
private void inOrderTraverse(TreeNode node) {
    if (node == null)
        return;
    inOrderTraverse(node.left);
    System.out.print(node.val);
    inOrderTraverse(node.right);
}

输出

test:12473568//47215386

转载于:https://www.cnblogs.com/aiguozou/p/11474786.html