剑指offer(32)II
剑指offer(32)II
剑指 Offer 32 - II. 从上到下打印二叉树 II
难度简单218
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
思路:
简单BFS,对上一题稍作更改即可。
class Solution {
public:
vector<vector<int>>ans;
vector<int>res;
vector<vector<int>> levelOrder(TreeNode* root) {
if(root==NULL)return ans;
//BFS
queue<TreeNode*>pq;//BFS核心数据结构
pq.push(root);
while(!pq.empty()){
//sz记录每一层的数量
int sz=pq.size();
//在记录的同时更新pq队列
for(int i=0;i<sz;i++){
TreeNode* cur=pq.front();
pq.pop();
res.push_back(cur->val);
if(cur->left!=NULL)pq.push(cur->left);
if(cur->right!=NULL)pq.push(cur->right);
}
//res记录的是每一层的数
ans.push_back(res);
res.clear();
}
return ans;
}
};
本文来自博客园,作者:{BailanZ},转载请注明原文链接:https://www.cnblogs.com/BailanZ/p/16205434.html