64.树 | 重建二叉树

最新推荐文章于 2024-11-11 20:03:56 发布

非社会人士

最新推荐文章于 2024-11-11 20:03:56 发布

阅读量218

点赞数

分类专栏：剑指offer 文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/weixin_43469527/article/details/130540305

版权

剑指offer 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

该文章介绍了一种方法，通过先序遍历和中序遍历的起始和结束下标，使用递归来构建二叉树。关键在于利用哈希映射存储中序遍历中每个值的下标，以快速找到中间节点，然后分别构造左右子树。这种方法优化了查找过程，提高了效率。

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

之前做过，参考16.树(5)中的最后一道题，用递归方式，将先序遍历的起始、结束下标，和中序遍历的起始、结束下标作为参数，在每个递归函数中先找到当前节点，即先序遍历的第一个节点，也就是中序遍历的某个中间节点，依据此节点将中序遍历一分为二，左边部分为当前节点的左子节点，右边部分为当前节点的右子节点，如此递归。

递归过程中要找到中间节点，如果每层递归中都用for循环来查找确定的话比较耗时，可以在一开始就用map保存每个val在inorder中的下标。

class Solution {
public:
    unordered_map<int, int> indexs;
    TreeNode* build(vector<int>& preorder, int beginPre, int endPre, vector<int>& inorder, int beginIn, int endIn) {
        if (beginPre > endPre) {
            return nullptr;
        }
        TreeNode* node = new TreeNode(preorder[beginPre]);
        int mid = indexs[preorder[beginPre]];
        int lenLeft = mid - beginIn, lenRight = endIn - mid;
        node->left = build(preorder, beginPre + 1, beginPre + lenLeft, inorder, beginIn, beginIn + lenLeft - 1);
        node->right = build(preorder, beginPre + lenLeft + 1, endPre, inorder, mid + 1, endIn);
        return node;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        int n = preorder.size();
        for (int i = 0; i < n; ++i) {
            indexs[inorder[i]] = i; 
        }
        return build(preorder, 0, n - 1, inorder, 0, n - 1);
    }
};