LeetCode-105-从前序与中序遍历序列构造二叉树

最新推荐文章于 2025-02-27 22:44:25 发布

原创最新推荐文章于 2025-02-27 22:44:25 发布 · 100 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

力扣专栏收录该内容

94 篇文章

订阅专栏

这篇博客介绍了如何通过前序遍历和中序遍历序列来构建二叉树。文章详细阐述了利用HashMap辅助找到根节点的思路，并提供了Java代码实现，包括递归的helper方法。此方法有效地将两个遍历序列转化为实际的二叉树结构。

LeetCode-105-从前序与中序遍历序列构造二叉树

在这里插入图片描述

思路

1.前序遍历可以知道根节点，然后在中序遍历中寻找根节点，根节点的左右分别为左右子树
2.为了方便寻找中序遍历的根节点，可以使用HashMap来存储

代码

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        Map<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<inorder.length;i++){
            map.put(inorder[i],i);
        }
        return helper(preorder,0,preorder.length,inorder,0,inorder.length,map);
    }
    TreeNode helper(int[] preorder,int p_s,int p_e,int[] inorder,int i_s,int i_e,Map<Integer,Integer> map){
            if (p_s == p_e)  return null;
            int v=preorder[p_s];
            TreeNode root=new TreeNode(v);
            int index=map.get(v);
            int len=index-i_s;
            root.left=helper(preorder,p_s+1,p_s+len+1,inorder,i_s,index,map);
            root.right=helper(preorder,p_s+len+1,p_e,inorder,index+1,i_e,map);
            return root;
    }
}