根据先序和中序序列还原二叉树递归剑指Offer 07.重建二叉树

最新推荐文章于 2022-04-13 09:51:02 发布

George_Will

最新推荐文章于 2022-04-13 09:51:02 发布

阅读量165

点赞数

分类专栏： LeetCode 文章标签：二叉树 dfs leetcode

本文链接：https://blog.csdn.net/weixin_44716674/article/details/108097878

版权

LeetCode 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

剑指Offer 07.重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

来源：LeetCode
网址：https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/

public TreeNode buildTree(int[] preorder, int[] inorder) {
    if(preorder == null || inorder == null || preorder.length != inorder.length || preorder.length == 0) return null;

    Map<Integer,Integer> inMap = new HashMap<>();
    for(int i = 0;i < inorder.length;i++){//便于根据preorder找到在inorder中的对应元素的下标
        inMap.put(inorder[i],i);
    } 

    return rebuildTree(preorder,inMap,0,preorder.length-1,0,inorder.length-1);
}

private TreeNode rebuildTree(int[] preorder,Map<Integer,Integer> inMap,int preL,int preR,int inL,int inR){
    if(preL > preR || inL > inR) return null;

    TreeNode node = new TreeNode(preorder[preL]);
    int index = inMap.get((Object)preorder[preL]);

    node.left = rebuildTree(preorder,inMap,preL+1,preL+index-inL,inL,index-1);
    node.right = rebuildTree(preorder,inMap,preL+1+index-inL,preR,index+1,inR);
    return node;
}