题目描述
从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。
样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
8
/ \
12 2
/
6
/
4
输出:[[8], [12, 2], [6], [4]]
问题分析
此题需要将各个层的节点分别保存到不同的数组中。所以在while循环中,加了一个for循环,循环次数是当前层的节点个数,这样虽然在for循环中不断的在队列尾部加入当前层节点的孩子节点,但是由于for循环限定了弹出栈的次数为当前层节点的个数,所以在for循环结束时,out中正好保存了当前层的所有节点,不多不少,同时栈中也只有下一层的所有节点,不多不少。然后在将out保存到ans中之后,进入下一次while循环,重复上述过程。最后当某一层不存在非空节点,即栈为空时,退出while循环。返回ans。
代码实现
/**
* 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) {
if(!root)
return {};
vector<vector<int>> ans;
queue<TreeNode*> s;
s.push(root);
while(!s.empty()){
vector<int> out;
for(int i = s.size(); i > 0; i--){
TreeNode* temp = s.front();
s.pop();
out.push_back(temp->val);
if(temp->left)
s.push(temp->left);
if(temp->right)
s.push(temp->right);
}
ans.push_back(out);
}
return ans;
}
};