JAVA-牛客-重建二叉树

12 篇文章 0 订阅
11 篇文章 0 订阅

重建二叉树

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

分析

通过前序遍历和中序遍历去重建二叉树
前序遍历代表根在最前方,中序遍历根在其左子树和右子树的中间。
我们通过创建根节点的方式去重建整个二叉树,
要用根来创建二叉树,需要对前序进行遍历,整个问题可以化分为数个子问题进行解决。
创建一个二叉树,首先创建根节点,其次创建左子树和右子树。
从上面数据,首先找到根节点->1,其在中序遍历的位置4,7,2,->1->5,3,8,6.
其左子树为4,7,2.右子树为5,3,8,6.
此时,由先序遍历可知,1的左子树为2,4,7.由此可以得出,2为1的左子根,
而1的右子根从前序遍历中在左子树的节点之后,右子根为3.
继续划分子问题,将先序遍历中1的左右子树也找出来,
左子树2,4,7,右子树为3,5,6,8
先将左子树按照上述方式进行创建,
2划分左右子树,4,7为2的左子树,在中序遍历中2的右边为1,因此没有右子树。
在先序遍历中2之后为4,且在中序遍历中4在2的左边,因此4为2的左子树,
又有7为4的子树,7在4的右边,7为4的右子树。
以此推导,按照根在先序遍历中的顺序来创建节点,按照中序遍历中的顺序来判断其是左子树还是右子树。

代码如下:

import java.util.Arrays;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length==0||in.length==0){
            return null;
        }
        TreeNode root = new TreeNode(pre[0]);
        buildTree(root,pre,in);
        return root;
        
    }
    public void buildTree(TreeNode root,int[] pre,int[] in){
        
        int i =0;
        for(;i<in.length;i++){
            if(in[i]==root.val){
                break;
            }
        }
        if(i>0){
            root.left = new TreeNode(pre[1]);
            buildTree(root.left,Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
        }
        if(in.length-i-1>0){
            root.right = new TreeNode(pre[i+1]);
            buildTree(root.right,Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
        }

        
        
    }
}

在这里插入图片描述

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值