【剑指Offer】4.重建二叉树

题目:

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

解答:

①利用Arrays里的copyOfRange函数
注意copyOfRange函数,Arrays.copyOfRange(oringinal,int from, int to)包含from的元素,不包含to的元素。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 //例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
import java.util.Arrays;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre==null||in==null){
            return null;
        }
        if(pre.length==0||in.length==0){
            return null;
        }
        if(pre.length!=in.length){
            return null;
        }
        TreeNode root=new TreeNode(pre[0]);
        for(int i=0;i<in.length;i++){
            if(pre[0]==in[i]){
                root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
//pre第0个元素就是这次的根,所以左子树为1开始,i是in的下标,是左子树节点的个数,i+1因为copyOfRange不包含后面的节点。in,0,i就是根节点左侧的左子树节点。                
                root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
//pre,右子树i+1包含在里面,到pre.length-1,pre.length不包含。in里面,i的位置是根,都不包含,从n+1开始。
                break;
            }
        }
        return root;
    }
}

运行时间: 261 ms
占用内存: 22744K

②拷贝数组会消耗大量内存,使用传入索引的方式
数字还在数组里面,只不过是通过索引的方式构造整个二叉树。

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
        return root;
    }
    //前序遍历{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
    private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn) {          
        if(startPre>endPre||startIn>endIn)
            return null;
        TreeNode root=new TreeNode(pre[startPre]);  
        for(int i=startIn;i<=endIn;i++)
            if(in[i]==pre[startPre]){
                root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
//pre和in一直没变过,startPre就是根节点,所以startPre+1开始是左子树的节点。i-startIn,i指的是中序遍历根节点的位置,减去中序遍历开始位置即左子树的长度。所以i-startIn+startPre是前序遍历左子树结束的位置。in就好理解了,startIn,i是根节点,中间就是左子树了。
                root.right=reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
//i-startIn跟上面的一样,+startPre就是根节点的位置,再+1是右子树。In同样道理
                      break;
            }

        return root;
    }
}

运行时间: 210 ms
占用内存: 21912K

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值