【C++牛客网刷题】-重建二叉树！

最新推荐文章于 2024-04-27 00:31:58 发布

Amazingren

最新推荐文章于 2024-04-27 00:31:58 发布

阅读量568

点赞数

分类专栏：牛客 C/C++学习

本文链接：https://blog.csdn.net/Amazingren/article/details/91620766

版权

C/C++学习同时被 2 个专栏收录

63 篇文章 0 订阅

订阅专栏

牛客

5 篇文章 1 订阅

订阅专栏

Task

在这里插入图片描述

思路：

解这个题之前我是根本不知道树的，
所以必须先要了解什么事前序遍历和中序遍历：
前序遍历：有左走左，无左走右，都无后退
中序遍历：只认准左孩子的左孩子…万不得已才后退，后退后也是上一个节点的右孩子，这样类推！
找了个说法：讲的还算透彻！
在这里插入图片描述

有了这个基础字后，就可以重建二叉树了！
后面的还是需要对树有一个深入的理解，要结合剑指Offer来思考和复习！这样快速一些！

代码

/**
 * 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) {
        int vinlen = vin.size();
        if(vinlen==0)
            return NULL;
        vector<int> left_pre, right_pre, left_vin, right_vin;
        //创建跟姐蒂娜，跟积淀肯定是前序遍历的第一个数
        TreeNode* head = new TreeNode(pre[0]);
        //找到中序遍历根节点所在位置，存放于变量gen中
        int gen = 0;
        for( int i=0; i<vinlen; i++)
        {
            if(vin[i] == pre[0])
            {
                gen = i;
                break;
            }
        }
        //对于中序遍历，根节点左边的节点位于二叉树的左边，根节点右边的节点位于二叉树的右边
        //利用上述这点，对二叉树归并
        for(int i=0; i<gen; i++)
        {
            left_vin.push_back(vin[i]);
            left_pre.push_back(pre[i+1]); //前序第一个为根节点
        }
        for(int i=gen+1; i<vinlen; i++)
        {
            right_vin.push_back(vin[i]);
            right_pre.push_back(pre[i]);
        }
        //和shell排序的思想类似，去除前序和中序遍历根节点左边和右边的子树
        //递归，再对其进行上述所有步骤，既再区分子树的左右子树，直到叶节点
        head->left=reConstructBinaryTree(left_pre, left_vin);
        head->right=reConstructBinaryTree(right_pre, right_vin);
        return head;
    }
};