1.先看下面这张图
层序遍历怎么工作的呢?
从左到右,丛上到下,依次遍历。
就和我们平时阅读的顺序是一样的。
第一行看完了,看第二行。
所以很明显,下面这个树的层序遍历序列是:
1,2,3,4,5,6,7,8,9
2.数据结构实现的逻辑
我们需要一个队列,具体流程如下图:
3.c++完整代码(超级详细的注释)
#include <vector>
#include <queue>
using namespace std;
class Solution{
public:
vector<vector<int>> LevelOrder(TreeNode *root){
queue<TreeNode*> que;//建立一个队列,这个队列里面放的元素是二叉树结点指针类型
if (root != NULL) que.push(root);//把树根放进去
vector<vector<int>> result;//result这个容器,存放最终层序遍历的顺序
while (!que.empty()) {
//只要队列不空,说明还有元素没遍历,那就一直循环
int size = que.size();//这里的que.size()就是上层元素个个数了
vector<int> vec;
//这个for循环,把每一层的元素都放在一个小列表vec里面,
//最后,再把一个个小列表加入到result这个列表里面就可以了
//所以这个函数的返回值类型是vector<vector<int>>类型
for (int i = 0; i < size; i++) {
//注意,这里i < size不能写成que.size(),因为que.size()在每一轮for循环后都会发生改变
//可能增大,也可能减小,程序容易出问题
//我们这里只要知道上一层有多少个元素,并且把上一层元素都遍历到就行了
//所以,在上面使用了int size = que.size(),提前把上一层元素的数量保存下来
//队头元素出队
TreeNode *node = que.front();//
que.pop();
vec.push_back(node->val);//元素放进vec列表的末尾
if (node->left) que.push(node->left);//若有左子树,进队
if (node->right) que.push(node->right);//若有有子树,进队
}
//每一层的的小列表加入到大列表里面去
result.push_bakc(vec);
}
return result;
}
}