LeetCode 102. Binary Tree Level Order Traversal

dongcl97

于 2020-04-07 17:46:36 发布

阅读量78

点赞数

分类专栏： LeetCode Tree 文章标签：二叉树数据结构 leetcode 算法

本文链接：https://blog.youkuaiyun.com/weixin_36634831/article/details/105369645

版权

LeetCode 同时被 2 个专栏收录

58 篇文章

订阅专栏

Tree

12 篇文章

订阅专栏

本文介绍了一种使用队列实现二叉树层序遍历的方法，通过记录每层节点数量，将遍历结果以二维数组形式输出，详细解析了算法思路与实现过程。

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

Problem

题目链接

Solution

给定一棵二叉树，输出对应的层序遍历结果。
二叉树层序遍历只需借助数据结构中的队列queue就可实现。
这题的输出要求按二维数组的格式输出，每层的数据对应于一维，那么某层的节点个数如何确定呢？我们可以在每层遍历开始时，记录一下这个时候队列的size，就是这一层的节点个数了。具体解释就是，每层遍历开始时，上层的节点已经全部出队了，下层的节点还未入队，因此当前的size就是这一层节点的个数。

/**
 * Definition for a binary tree node.
 * 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) {
        vector<vector<int> > ans;
        queue<TreeNode*> Q;
        if(root==NULL) return ans;
        Q.push(root);
        while(!Q.empty())
        {
            int s=Q.size();
            vector<int> tv;
            for(int i=0;i<s;++i)
            {
                TreeNode* p=Q.front();
                tv.push_back(p->val);
                if(p->left) Q.push(p->left);
                if(p->right) Q.push(p->right);
                Q.pop();
            }
            ans.push_back(tv);
        }
        return ans;
    }
};