前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >☆打卡算法☆LeetCode 106、从中序与后序遍历序列构造二叉树 算法解析

☆打卡算法☆LeetCode 106、从中序与后序遍历序列构造二叉树 算法解析

作者头像
恬静的小魔龙
发布2022-08-07 10:25:27
1710
发布2022-08-07 10:25:27
举报
文章被收录于专栏:Unity3DUnity3D
大家好,我是小魔龙,Unity3D软件工程师,VR、AR,虚拟仿真方向,不定时更新软件开发技巧,生活感悟,觉得有用记得一键三连哦。

一、题目

1、算法题目

“给定两个整数数组ino和pos,其中ino是二叉树的中序遍历,pos是二叉树的后序遍历,请你构造并返回这颗二叉树。”

题目链接:

来源:力扣(LeetCode)

链接:106. 从中序与后序遍历序列构造二叉树 - 力扣(LeetCode) (leetcode-cn.com)

2、题目描述

给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。

image.png
image.png
代码语言:javascript
复制
示例 1:
输入:inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出:[3,9,20,null,null,15,7]
代码语言:javascript
复制
示例 2:
输入: inorder = [-1], postorder = [-1]
输出: [-1]

二、解题

1、思路分析

先来了解一下什么是中序遍历和后序遍历:

  • 中序遍历的顺序是先遍历左子树,再遍历根节点,最后遍历右子树
  • 后序遍历的顺序是先遍历左子树,再遍历右子树,最后遍历根节点

根据中序遍历和后序遍历的性质,我们可以知道后序遍历的数组最后一个元素就是根节点。

根据这个性质,我们可以使用根节点的信息在中序遍历的数组中找到根节点所在的下标。

然后根据其在中序遍历的数组分成左右两部分,就是左右子树,然后同样的方法递归递归构造下去。

2、代码实现

代码参考:

代码语言:javascript
复制
class Solution {
    int post_idx;
    int[] postorder;
    int[] inorder;
    Map<Integer, Integer> idx_map = new HashMap<Integer, Integer>();

    public TreeNode helper(int in_left, int in_right) {
        // 如果这里没有节点构造二叉树了,就结束
        if (in_left > in_right) {
            return null;
        }

        // 选择 post_idx 位置的元素作为当前子树根节点
        int root_val = postorder[post_idx];
        TreeNode root = new TreeNode(root_val);

        // 根据 root 所在位置分成左右两棵子树
        int index = idx_map.get(root_val);

        // 下标减一
        post_idx--;
        // 构造右子树
        root.right = helper(index + 1, in_right);
        // 构造左子树
        root.left = helper(in_left, index - 1);
        return root;
    }

    public TreeNode buildTree(int[] inorder, int[] postorder) {
        this.postorder = postorder;
        this.inorder = inorder;
        // 从后序遍历的最后一个元素开始
        post_idx = postorder.length - 1;

        // 建立(元素,下标)键值对的哈希表
        int idx = 0;
        for (Integer val : inorder) {
            idx_map.put(val, idx++);
        }
        
        return helper(0, inorder.length - 1);
    }
}
image.png
image.png

3、时间复杂度

时间复杂度 : O(n)

其中n是树中的节点个数。

空间复杂度: O(n)

其中n是树中的节点个数。

三、总结

为了高效地查找根节点元素在中序表遍历数组中的下标,我们可以使用哈希表来存储中序序列。

在递归的过程中,利用哈希表(1)的时间复杂度查询当前根节点在中序遍历中的下标。

根绝后序遍历性质,递归创建右子树和左子树,创建左右子树的依赖关系,再存储右子树的节点,最后存储根节点。

返回根节点root。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-02-21,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 一、题目
    • 1、算法题目
      • 2、题目描述
      • 二、解题
        • 1、思路分析
          • 2、代码实现
            • 3、时间复杂度
            • 三、总结
            领券
            问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档