题目描述
从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。
数据范围
树中节点的数量 $[0,1000]$。
样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
8
/ \
12 2
/
6
/
4
输出:[[8], [12, 2], [6], [4]]
思路
这道题本质上还是二叉树的层序遍历,但由于普通的层序遍历不同,普通的$bfs$是每次取一个结点拓展,而这道题我们可以每次取队列中的所有数进行拓展,就能把下一行的所有数都求出来。
代码
/**
* 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>> printFromTopToBottom(TreeNode* root) {
vector<vector<int>> res;//答案
queue<TreeNode*> q;//队列
if (root == nullptr)//SF:喵喵喵?
return {};
q.push(root);
while (!q.empty())
{
vector<int> level;//临时数组
int l = q.size();//保存当前队列的长度,因为后面会删除和添加一些数
for (int i = 0; i < l; i ++)
{
//板子
auto t = q.front();
q.pop();
level.push_back(t -> val);
if (t -> left)
q.push(t -> left);
if (t -> right)
q.push(t -> right);
}
//把临时数组加入答案里去
res.push_back(level);
}
return res;
}
};
好了,这篇题解到这里就结束了。感谢观看!
ヽ( ̄ω ̄〃)ゝ
$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\quad$$\mathcal{writer\enspace by \enspace acwing}$ : $\mathfrak{天元之弈}$