算法学习-06-重构二叉树

最新推荐文章于 2024-01-27 19:52:14 发布

新亲斤手

最新推荐文章于 2024-01-27 19:52:14 发布

阅读量152

点赞数

分类专栏：算法总结文章标签：二叉树 c++ 算法数据结构 leetcode

本文链接：https://blog.csdn.net/qq_41443458/article/details/113400896

版权

算法总结专栏收录该内容

6 篇文章 0 订阅

订阅专栏

算法学习-06-重构二叉树-C++

题目：输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。
输入：[1,2,3,4,5,6,7],[3,2,4,1,6,5,7]
输出：{1,2,5,3,4,6,7}
分析：已知二叉树的先序和中序遍历，要求重构二叉树。先序的顺序是：（1）当前节点，（2）当前节点的左子节点，（3）当前节点的右子节点。中序遍历的顺序是：（1）当前节点的左子节点，（2）当前节点，（3）当前节点的右子节点。可以更具先序和中序的特点来重构二叉树，详细过程请看重建二叉树

C++代码实现

class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size() == 0 || vin.size() == 0) return nullptr;
        
        return re(pre, vin, 0, pre.size() - 1, 0, vin.size() - 1);
    }
    
    TreeNode* re(vector<int>& pre, vector<int>& vin, int p1, int p2, int v1, int v2){
        if(p2 < p1 || v2 < v1) return nullptr;
        int size;
        
        TreeNode* node = new TreeNode(pre[p1]);
        for(int i = v1; i <= v2; i++){
            if(pre[p1] == vin[i]){
                size = i - v1;
                break;
            }
        }
        node->left = re(pre, vin, p1 + 1, p1 + size, v1, v1 + size - 1);
        node->right = re(pre, vin, p1 + size + 1, p2, v1 + size + 1, v2);
        
        return node;
    }