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

最新推荐文章于 2024-04-07 15:47:16 发布

DataPlayerK

最新推荐文章于 2024-04-07 15:47:16 发布

阅读量90

点赞数 1

分类专栏：算法文章标签： 1024程序员节二叉树递归法递归算法

本文链接：https://blog.csdn.net/qq_50757994/article/details/120930743

版权

算法专栏收录该内容

317 篇文章 1 订阅

订阅专栏

这篇博客详细介绍了如何通过前序和中序遍历序列来构造二叉树的问题。文章指出，这是一个递归过程，首先找到前序遍历中的根节点，然后根据中序遍历中根节点的位置划分左右子树，并递归构建。代码实现部分展示了如何利用哈希映射辅助找到节点位置，从而高效地构建二叉树结构。

摘要由CSDN通过智能技术生成

题目描述

点这里

思路分析

二叉树的遍历构造问题
本质是递归问题。前序遍历的第一个点即为根节点，再根据中序遍历中根节点的位置确定左右子树的节点个数。然后递归处理左右子树。

代码实现

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    unordered_map<int,int> hash;
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        for(int i=0;i<inorder.size();i++) hash[inorder[i]]=i;
        return dfs(preorder,inorder,0,preorder.size()-1,0,inorder.size()-1); 
    }
    TreeNode* dfs(vector<int>& preorder, vector<int>& inorder,int pl,int pr,int il,int ir){
        if(pl>pr) return NULL;
        auto root=new TreeNode(preorder[pl]);
        int k=hash[root->val];
        root->left=dfs(preorder,inorder,pl+1,pl+1+k-1-il,il,k-1);
        root->right=dfs(preorder,inorder,pl+1+k-il,pr,k+1,ir);
        return root;
    }
};