&LeetCode102& 二叉树的层次遍历

题目

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

提示：
节点总数 <= 1000

来源：力扣（LeetCode）

思路

首先，可以建立一个 queue（队列）；
然后，把根节点放入，这时候找根节点的左右两个子节点，此时去掉根节点，即 queue 里的元素就是下一层的所有节点；
其次，用一个 for 循环遍历它们，然后存到一个一维向量里，遍历完之后再把这个一维向量存到二维向量里；
最后，以此类推，可以完成层序遍历。

C++代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) 
    {
        if (!root)
            return {};
        vector<vector<int>> res;
        queue<TreeNode*> q{{root}};
        while (!q.empty())
        {
            vector<int> oneLevel;
            for (int i = q.size(); i > 0; --i)
            {
                TreeNode *t = q.front();
                q.pop();
                oneLevel.push_back(t -> val);
                if (t -> left)
                    q.push(t -> left);
                if (t -> right)
                    q.push(t -> right);
            }
            res.push_back(oneLevel);
        }
        return res;
    }
};