LeetCode level-order bfs

weixin_42246923

于 2018-06-02 10:39:35 发布

阅读量132

点赞数

CC 4.0 BY-SA版权

分类专栏： bfs

本文链接：https://blog.youkuaiyun.com/weixin_42246923/article/details/80545911

bfs 专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种使用队列实现二叉树层次遍历的方法，并通过C++代码示例展示了如何将各层节点值存储到二维数组中。此算法适用于需要按层获取二叉树节点的应用场景。

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

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
        queue<TreeNode*> q;
        vector<vector<int>> res;
        if(root==NULL) return res;
        int cur,next;
        next = 0;  //表示层数的方法1，可以再结构体中加入一个变量表示当前的层数
        cur = 1;
        q.push(root);
        vector<int> tmp;
        while(!q.empty()){
            TreeNode *p = q.front();
            q.pop();
            cur--;
            tmp.push_back(p->val);
            if(p->left){
                next++;
                q.push(p->left);
            }
            if(p->right){
                next++;
                q.push(p->right);
            }
            if(cur==0){
                res.push_back(tmp);
                tmp.clear();
                cur = next;
                next = 0;
            }
        }
        return res;
    }
};