由前序和中序数组重建二叉树

35 篇文章 0 订阅
15 篇文章 0 订阅

1、重建二叉树
来源:牛客网

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

2、思路:先序第一个元素a为根节点,在中序数组中,a左边的元素为左子树,右边为右子树。左右子树放到递归中处理即可

3、代码:

private TreeNode conBinaryTreeByPreAndIn(TreeNode root, int[] pre, int[] in, int pre_start,
            int pre_end, int in_start, int in_end) {

        int i, in_pos = 0;
        // 找到先序中第一个元素在中序中的下标pos
        for (i = in_start; i <= in_end; i++) {
            if (in[i] == root.val) {
                in_pos = i;
                break;
            }
        }
        TreeNode leftNode = null, rightNode = null;
        int left_num = in_pos-in_start;//左子树数目

        // 当前根节点存在左子树
        if (in_pos > in_start) {
            leftNode = new TreeNode(pre[pre_start + 1]);
            root.left = leftNode;
        }
        // 当前根节点存在右子树
        if (in_pos < in_end) {
            rightNode = new TreeNode(pre[pre_start+(left_num+1)]);
            root.right = rightNode;
        }

        // 根节点左边子节点数大于1
        if (in_pos - in_start > 1) {
            conBinaryTreeByPreAndIn(leftNode, pre, in, pre_start + 1, pre_start + left_num,
                                             in_start , in_pos - 1);
        }
        // 根节点右边子节点数大于1
        if (in_end - in_pos > 1) {
            conBinaryTreeByPreAndIn(rightNode, pre, in, pre_start + left_num + 1, pre_end,
                                             in_pos + 1, in_end);
        }
        return root;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值