JZ7 重建二叉树

最新推荐文章于 2024-11-02 16:10:25 发布

GalaxySummer

最新推荐文章于 2024-11-02 16:10:25 发布

阅读量102

点赞数

分类专栏：剑指offer 文章标签： c++

本文链接：https://blog.csdn.net/pure_dreams/article/details/124431091

版权

剑指offer 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

这篇博客主要介绍了如何通过前序遍历和中序遍历的序列来重建二叉树的算法。核心思路是找到根节点，然后根据中序遍历中根节点的位置划分左右子树，采用递归方式分别构造左右子树。代码实现使用了C++，并强调了分治法的空间范围切割和递归调用过程。

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

题目链接:重建二叉树_牛客题霸_牛客网

注意点：

找到前序遍历的根节点，然后再确定中序遍历中根节点的左右子树。

切割空间的范围用的是分治法，按照左开右闭的原则，多写多调试吧。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
    if(pre.empty() || vin.empty()) return NULL;
    //建立根节点
    int rootval = pre[0];
    TreeNode* root = new TreeNode(rootval);
    //找出根节点在中序遍历中的位置，确定左右节点子树个数
    int pos = 0;
    for(; pos < vin.size(); ++pos){
        if(vin.at(pos) == root->val) break;
    }
    //如果中序没有根节点
    //if (vin[pos] == vin.end()) break;
    //vector的拷贝构造函数，后续递归调用时需要用到的四个左右子树vector区间
    vector<int> left_pre(pre.begin()+1, pre.begin()+pos+1);
    vector<int> left_vin(vin.begin(), vin.begin()+pos);
    vector<int> right_pre(pre.begin()+pos+1, pre.end());
    vector<int> right_vin(vin.begin()+pos+1, vin.end());
    //递归构造左右子树
    root->left = reConstructBinaryTree(left_pre, left_vin);
    root->right = reConstructBinaryTree(right_pre, right_vin);
    return root;
    }
};