leetcode+二叉树层次遍历+二维vector使用

最新推荐文章于 2024-06-26 21:21:29 发布

小堃哥

最新推荐文章于 2024-06-26 21:21:29 发布

阅读量315

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/u013554860/article/details/79598245

Leetcode 专栏收录该内容

450 篇文章

订阅专栏

这是一个使用C++解决LeetCode二叉树层次遍历问题的示例，通过队列和二维vector实现。代码中定义了一个Solution类，包含levelOrder方法，该方法遍历树的每一层，将节点值存入vector中，并返回结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string>
#include<cstring>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#include<limits.h>
#include<assert.h>
#include<sstream>
#define LL long long
#define inf 0x3f3f3f3f
#define mod 1e9+7
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
/*
 class Solution {
    struct Node
    {
        int flag;
        TreeNode* r;
    };
public:
    vector< vector<int> > levelOrder(TreeNode* root) {
        vector< vector<int> > res;
        if(!root) return res;
        ///res[0].push_back(root->val);
        queue<Node> Q;
        Node temple1, temple2;
        //TreeNode* left1=NULL, right1=NULL;
        temple1.flag=0; temple1.r=root;
        Q.push(temple1);
        while (!Q.empty()) {
            temple1=Q.front(); Q.pop();
            if(temple1.flag>=0) res[temple1.flag].push_back(temple1.r->val);
            if(temple1.r->left!=NULL){
                temple2.flag =temple1.flag+1;
                temple2.r = temple1.r->left;
                Q.push(temple2);
            }
            if(temple1.r->right!=NULL){
                temple2.flag = temple2.flag+1;
                temple2.r= temple2.r->right;
                Q.push(temple2);
            }
        }
        return res;
    }
};
*/
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if(!root) return res;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()){
            vector<int> tmpRes; //二维vector中存储每一层的vector
            queue<TreeNode*> tmpQue; //临时的队列，为了消耗每一层队列中的值
            while(!que.empty()){
                TreeNode *node = que.front();
                que.pop();
                if(node->left) tmpQue.push(node->left);
                if(node->right) tmpQue.push(node->right);
                tmpRes.push_back(node->val);//每一层vector的值
            }
            res.push_back(tmpRes);
            que = tmpQue; //赋值一下
        }
        return res;
    }
};
int main()
{
    Solution temple;
    return 0;
}

点击打开链接