LeetCode 从中序遍历和后序遍历构建二叉树（java）

最新推荐文章于 2024-10-25 16:10:03 发布

csu_175

最新推荐文章于 2024-10-25 16:10:03 发布

阅读量1.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：数据结构二叉树中序遍历后序遍历还原二叉树

本文链接：https://blog.youkuaiyun.com/qq_37174705/article/details/88864530

LeetCode 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种通过给定的中序遍历与后序遍历序列构造二叉树的方法。利用哈希表快速定位根节点，并递归地构建左右子树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树：

思路：

①后续遍历的最后一个数为根结点，根据这个根结点来划分中序遍历，将其分为左子树和右子树

②确定左右子树的中序遍历和后续遍历中的界限

③递归调用

代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        if(inorder.length == 0 || postorder.length == 0 || inorder.length != postorder.length){
            return null;
        }
        java.util.Map<Integer,Integer> map = new java.util.HashMap<>();
        //将中序遍历的node和下标进行映射
        for(int i = 0;i < inorder.length;i++){
            map.put(inorder[i],i);
        }
        return buildNode(inorder,0,inorder.length-1,postorder,0,postorder.length-1,map);
    }
    
    public TreeNode buildNode(int[] inorder,int begin1,int end1, 
                              int[] postorder,int begin2,int end2,
                              java.util.Map<Integer,Integer> map) {
         if(begin2 > end2) {
        	return null;
        }
        if(begin1 == end1){
            return new TreeNode(inorder[begin1]);
        }
        TreeNode root = new TreeNode(postorder[end2]);
        int index = map.get(postorder[end2]);
        root.left = buildNode(inorder,begin1,index-1,postorder,begin2,begin2+index-begin1-1,map);
        root.right = buildNode(inorder,index+1,end1,postorder,begin2+index-begin1,end2-1,map);
        return root;
    }
}