题目描述
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
问题分析
此题需要将各个层的节点分别保存到不同的数组中。所以在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>> levelOrder(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;
}
};