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

题目地址：力扣

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

class Solution {
public:
    vector> levelOrder(TreeNode* root) {
        // 结果数组
        vector> res;
        // 边界条件，树为空
        if (root != nullptr)
        {
            // 初始化一个队列存储根节点
            queue treeq;
            treeq.push(root);
            // 只要队列不空就继续循环
            while (!treeq.empty())
            {
                // 当前队列的长度就代表本层节点的个数，因为我们需要按层输出所以要记录下每层个数
                int cnt = treeq.size();
                // 用于存储本层节点的值
                vector 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;
    }
};

Accepted

• 34/34 cases passed (4 ms)
• Your runtime beats 70.9 % of cpp submissions
• Your memory usage beats 78.34 % of cpp submissions (12.1 MB)