二叉树的层次遍历
题目:从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
链接:剑指Offer32-从上到下打印二叉树
思路:
关键点:二叉树,按层。可以看出考察的是二叉树的广度优先遍历。
问题:对二叉树的几种遍历的框架不熟。没办法了,先翻书吧。先简化,把每一步的逻辑理清楚。
工具:广度优先遍历利用了队列先入先出的特性。
数据结构:总的结果放在一个向量res里,每一层的结果放在一个向量tmp里(局部的,每一层要更新)。有一个队列que用来遍历。
队列中只会保存一层的节点。
两层循环:循环每一层,循环一层中的每一个节点。
对每个节点的处理:打印在层向量tmp中以及将左右孩子入队列。
对每一个层向量的处理:不为空时,打印在结果向量res中。
循环每一层的节点时,如果为空节点则退出继续下一次循环。
代码:
vector<vector<int>> levelPrint(TreeNode* root)
{
//先摆出工具
queue<TreeNode*> que;
vector<vector<int>> res;
que.push(root);
//开始第一层循环:遍历每一层。结果在res。
//在打印结果前,要判断每一层的tmp是否为空
while(que.size())
{
int size = que.size();
vector<int> tmp;
//开始第二层循环:遍历一层中每个节点。结果在tmp。
//在打印结果前,要判断每个节点是否为空。
for(int i=0;i<size;i++)
{ //一层中对每个节点要做的处理
TreeNode* rt = que.front();
que.pop();
if(!rt) continue;
tmp.push_back(rt->val);
if(rt->left) que.push(rt->left);
if(rt->right) que.push(rt->right);
}
if(tmp.size()!=NUll) res.push_back(tmp);
}
return res;
}