首页 > 其他 > 详细

重建二叉树

时间:2019-09-06 15:56:37      阅读:76      评论:0      收藏:0      [点我收藏+]

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!