给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
本题我是利用BFS遍历来实现,按层将节点不断压入队列,因为同一层的节点是在一个vector中的。所以还需要记录节点所在层数。这里恰好可以利用“当前层的节点个数为队列中元素个数”这个性质。
代码展示:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> ans;
queue<TreeNode*> q;
if(root==NULL)
return ans;
q.push(root);
while(!q.empty()){
vector<int> temp;
int size = q.size();
while(size>0){
TreeNode* node = q.front();
temp.push_back(node->val);
q.pop();
if(node->left!=NULL)
q.push(node->left);
if(node->right!=NULL)
q.push(node->right);
size -= 1;
}
ans.push_back(temp);
}
return ans;
}
};