力扣 102 二叉树的层序遍历【广度优先遍历】
全部刷题与学习记录
原题目
题目地址:102. 二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
注:本题与 剑指 32_2 分行从上到下打印二叉树【广度优先遍历】【层序遍历】 相同,属于二刷,但是做起来还是没有那么顺手,因为对与队列的处理不太娴熟
考查知识点
层序遍历、广度优先遍历
自己的第一遍解法
层序遍历的代码是比深度优先难想一点的,因为用到了一个队列,并且需要按行区分数组
要点就是在访问队列当前队首节点时,要将访问节点的子节点(如果有的话)加入队列,放在队尾。因为要按行区分数组,所以要有一个计数的概念,要在一行所有节点都加入到一个数组中以后,自动开始下一数组。主要是这个计数器的实现,其实用两层循环就好了
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if (root == nullptr) return {
};
queue<TreeNode*> que; //逐步存储某一层全部节点
que.push(root);
vector<vector<int>> res;
while (!que.empty()) {
vector<int> level; //当前层所有数字
int size = que.size();
//将队列中(当前层)所有数字添加到数组中,自动形成分数组存储不同层
for (int i = 0; i < size; ++i) {
TreeNode* cur = que.front();
que.pop();
level.push_back(cur->val);
if (cur->left) que.push(cur->left);
if (cur->right) que.push(cur->right);
}
res.push_back(level);
}
return res;
}
};
参考:
https://mp.weixin.qq.com/s/Gb3BjakIKGNpup2jYtTzog