剑指offer——重建二叉树

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

首先我们要懂怎么重新建二叉树,在用代码实现,我当时做到题目想到的几个点。1,前序遍历第一个数为根节点。2,在中序遍历中找出根节点,根节点左边为左子树,右边为右子树,可以使用递归,拼接完左子树,在拼右子树。3.怎么才能知道前序遍历到第几个数为止是它的左子树呢?

4.怎么才能知道前序遍历根节点后面的那个结点是左孩子还是右孩子呢?5.取出根节点,递归遍历左子树,递归遍历右子树。水平不够,没能解出来,以为这种方法不行。下面来解析代码。个人理解,有错误请指出。

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
               if(pre==null||in==null){
            return null;
        }
 
        java.util.HashMap<Integer,Integer> map= new java.util.HashMap<Integer, Integer>();
        for(int i=0;i<in.length;i++){
            map.put(in[i],i);
        }//将中序遍历数组的键,值存入到Hashmap中
        return preIn(pre,0,pre.length-1,in,0,in.length-1,map);
    }
     
      public TreeNode preIn(int[] p,int pi,int pj,int[] n,int ni,int nj,java.util.HashMap<Integer,Integer> map){
         //传入参数,为什么要这么传?后面可以看到,这样完美解决了上面3,4两点问题。
        if(pi>pj){
            return null;
        }
        TreeNode head=new TreeNode(p[pi]);//pi初始值0,也就是拿到根节点
        int index=map.get(p[pi]);//找到根结点在中序遍历数组中的位置
        head.left=preIn(p,pi+1,pi+index-ni,n,ni,index-1,map);//递归,如果有左孩子,根节点指向它
        head.right=preIn(p,pi+index-ni+1,pj,n,index+1,nj,map);//递归,如果有右孩子,根结点指向它
        return head;
    }
}
首先,这个函数是怎么解决第3点问题的。进入函数之后,我们拿到根节点和根节点在中序遍历数组中的位置。这时候,在中序遍历数组中,0到Index-1就是根结点的左子树,这样我们就可以解决前序遍历中左子树是到哪个结点结束的,就是一个最简单的数学问题了,Index-1-ni=x-(pi+1),得到根节点的左子树的最后一个叶子结点的位置。取出来的根节点就不要在管它了,递归。

当pj<pi的时候,也就是该结点没有左孩子了,返回Null,比如当指针指向4的时候,4只有孩子没有左孩子,这样第一个递归结束。这是找到每个结点的左孩子。找到右孩子同理。

我们往根节点的右边找,中序遍历中index+1为右子树的起始位置,nj为length-1(因为是往右边在找)为右子树的结束位置,pj是固定好为length-1也是右子树结束的位置。这样得到根节点右孩子的位置,解决了第4点问题。判断下一个结点是为左孩子还是右孩子。

把中序遍历当作一个参照数组,可以理解得清楚一点。

关于递归的问题,还是得在自己得脑子里面跑一遍才能理清。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值