LeetCode 102. 二叉树的层序遍历(C++)

Xaiver_97

已于 2022-09-09 22:44:38 修改

阅读量462

点赞数

分类专栏：力扣刷题文章标签： c++ leetcode 算法

于 2022-09-09 22:19:17 首次发布

本文链接：https://blog.csdn.net/xavier_97/article/details/126789977

版权

力扣刷题专栏收录该内容

42 篇文章 2 订阅

订阅专栏

题目地址：力扣

思路：二叉树的层序遍历要求是一层一层输出节点，返回值应该是一个二维数组。我们可以用一个队列来存储每一层的节点，输出节点的时候将其左孩子和右孩子加入队列。同时使用一个变量来记录每一层的节点数量，当出队的节点数量等于该层节点数，就完成了本层的遍历。以此类推，直到队列为空就说明遍历完了整棵树。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        // 结果数组
        vector<vector<int>> res;
        // 边界条件，树为空
        if (root != nullptr)
        {
            // 初始化一个队列存储根节点
            queue<TreeNode*> treeq;
            treeq.push(root);
            // 只要队列不空就继续循环
            while (!treeq.empty())
            {
                // 当前队列的长度就代表本层节点的个数，因为我们需要按层输出所以要记录下每层个数
                int cnt = treeq.size();
                // 用于存储本层节点的值
                vector<int> curLayer;
                // 循环，使得刚好出队cnt个节点
                while (cnt != 0)
                {
                    // 将队头节点取出，将其左右孩子依次加入队列（若非空）
                    TreeNode* cur = treeq.front();
                    treeq.pop();
                    if (cur->left != nullptr)
                        treeq.push(cur->left);
                    if (cur->right != nullptr)
                        treeq.push(cur->right);
                    // 当前节点值加入当前层数组
                    curLayer.push_back(cur->val);     
                    --cnt;                   
                }
                // 将当前层遍历的结果数组加入整个树的结果数组
                res.push_back(curLayer);
            }
        }
        // 返回结果数组
        return res;
    }
};