重建二叉树

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

思路:

记前序遍历pre[],中序遍历in[].

先获取当前前序遍历数组pre的第一个元素记作a,比如遍历到i位置的值为K,在中序遍历in的(start-end区间)中找到和k值相等的元素,所在的位置为L。

则以i为父节点的左边子树的元素在in中 (start,i-1),左子树中前序遍历的第一个元素位置则为a+1;

以i为父节点的右边子树的元素在in中(i+1,end),右子树中前序遍历的第一个元素位置则为a+1+(i-start)

我们只需要通过递归重复两步就能将树构造出来。

需要注意的是递归的结束条件是:

左边节点left>=right或者中序遍历的第一个元素超出了数组范围。

public TreeNode reConstructBinaryTree(int[] pre,int[] in) {
        return construct(pre,in,0,0,in.length-1);
    }

public TreeNode construct(int[] pre,int[] in,int prestart,int instart,int inend){
            if(prestart>pre.length||instart>inend){
            return null;
        }else if(instart==inend){
            return new TreeNode(in[instart]);
        }
        int middle = pre[prestart];
        int temp = -1;
        for(int i=instart;i<=inend;i++){
            if(in[i]==middle){
                temp=i;
                break;
            }
        }
        TreeNode t = new TreeNode(middle);
        t.left=construct(pre,in,prestart+1,instart,temp-1);
        t.right=construct(pre,in,prestart+1+(temp-instart),temp+1,inend);
        return t;
    }

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值