剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:
在这里插入图片描述

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1]
Output: [-1]
限制:
0 <= 节点个数 <= 5000
思路
本题主要用到了二叉树前序和中序遍历的特性,前序遍历,每次遍历,根节点一定在第一位,中序遍历,每次遍历,根节点一定在中间,能够划分左右子树,因此每次可以通过前序遍历的第一位,得知根节点的值,再从中序遍历的结果中找到根节点,从而划分左右子树
代码

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
    }

    public TreeNode build(int[] preorder, int s1, int e1, int[] inorder,int s2,int e2) {
        if(s1>e1||s2>e2||s1>=preorder.length||s2>=inorder.length){
            return null;
        }
        TreeNode root = new TreeNode(preorder[s1]);
        int idx = s2;
        int root_num = preorder[s1];
        for(int i=s2;i<=e2;i++){
            if(inorder[i]==root_num){
                idx = i;
                break;
            }
        }
        int len1 = idx-1-s2;
        int len2 = e2-idx-1;
        root.left = build(preorder,s1+1,s1+idx-s2,inorder,s2,idx-1);
        root.right = build(preorder,s1+idx-s2+1,e1,inorder,idx+1,e2);
        return root;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值