【算法】leetcode 105 从前序与中序遍历序列构造二叉树

题目

输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 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

解答

cpp 复制代码
#include <iostream>
#include <unordered_map>
#include <vector>

using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
       for (int i = 0; i < inorder.size(); i++)
        {
            // 用中序遍历数组建立 值-下标 的映射
            value_index[inorder[i]] = i;
        }
        return recursive(preorder,0, 0, inorder.size() - 1);
    }
    TreeNode *recursive(vector<int>& pre,int pre_root, int in_left, int in_right)
    {
        if (in_left > in_right) return nullptr;
        // 将对应的 val 赋给 node 节点
        TreeNode *node = new TreeNode(pre[pre_root]);
        int in_root = value_index[pre[pre_root]];
        node->left = recursive(pre,pre_root + 1, in_left, in_root - 1);
        node->right = recursive(pre,pre_root + in_root - in_left + 1, in_root + 1, in_right);
        return node;
    }
private:
    unordered_map<int, int> value_index;
};
相关推荐
ゞ 正在缓冲99%…几秒前
leetcode375.猜数字大小II
数据结构·算法·leetcode·动态规划
Greedy Alg4 分钟前
LeetCode 79. 单词搜索
算法
碧海银沙音频科技研究院16 分钟前
i2s的LRCK时钟有毛刺以及BCLK数据在高采样率有变形数据解析错误问题原因以及解决方法
人工智能·深度学习·算法·分类·音视频
kida_yuan1 小时前
【从零开始】17. 中文摘要提取工具
python·算法·数据分析
水蓝烟雨1 小时前
0430. 扁平化多级双向链表
数据结构·链表
未到结局,焉知生死2 小时前
PAT每日三题
算法
阿巴~阿巴~2 小时前
Linux线程与进程的栈管理、页表机制及线程封装
数据结构·线程·进程·线程封装·页表机制·栈管理
进击的炸酱面2 小时前
第三章 线性模型
人工智能·算法·机器学习
立志成为大牛的小牛2 小时前
数据结构——三十一、最小生成树(王道408)
数据结构·学习·程序人生·考研·算法
CoovallyAIHub2 小时前
一致性模型:单步生成高质量图像,破解扩散模型速度瓶颈
深度学习·算法·计算机视觉