重建二叉树- 力扣

最新推荐文章于 2024-10-31 16:16:13 发布

久菜

最新推荐文章于 2024-10-31 16:16:13 发布

阅读量78

点赞数

分类专栏：二叉树文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_52669146/article/details/130073229

版权

二叉树专栏收录该内容

5 篇文章 0 订阅

订阅专栏

该代码实现了一个C++类Solution，包含两个方法用于根据给定的前序遍历和中序遍历（或后序遍历和中序遍历）序列重建二叉树。主要使用了哈希映射存储中序遍历中的节点位置，然后通过递归方式构建树的左右子节点。

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

题目详情

前序+中序：

后序+中序也是一个道理：

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 TreeRecur(0,0,preorder.size()-1,preorder);
    }
    TreeNode* TreeRecur(int rooti,int lefti,int righti,vector<int>& preorder){
        if(lefti > righti)//等于的时候不要退出，说明是叶子结点，需要构建
            return nullptr;

        TreeNode* root = new TreeNode(preorder[rooti]);
        int index = hash[preorder[rooti]];
        root->left = TreeRecur(rooti+1,lefti,index-1,preorder);
        root->right = TreeRecur(rooti+1+(index-lefti),index+1,righti,preorder);
        return root;
    }
};