算法笔记
之前谈到了二叉树的前中序遍历,也给出了非递归与递归两种情况下的代码,本次算法笔记将继续学习一种遍历方法,层序遍历。层序遍历可以理解为广度优先遍历,即从左到右遍历每一层节点即可。
题目描述
解题过程
对于层序遍历,我们需要使用数据结构队列去实现,因为层序遍历的过程符合队列先进先出的特点。拿题目描述中的例子举例,先将根节点3放入队列中,要实现层序遍历就先弹出节点3,记录3,然后它的子节点9和20再进入队列以此类推,弹出相应节点,放入他们的子节点,知道遍历完即可。
所以,在写代码的时候,我们可以 先定义数据结构队列与记录结果的二维数组result ,如果存在头结点就将其放入队列,之后使用循环,判断条件为队列是否为空, 定义size为队列长度,也可以不定义直接用也可以,再来一个一维数组path记录每一层的节点。 这里再来一个for循环,弹出头结点,记录进path数组,然后在队列中加入子节点即可,整部分代码就可以实现了。
代码实现
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> queue;
vector<vector<int>> result;
if(root != nullptr) {
queue.push(root);
}
while(!queue.empty()) {
int size = queue.size();
vector<int> path;
for(int i = 0; i < size; i++) {
TreeNode* node = queue.front();
queue.pop();
path.push_back(node -> val);
if(node -> left) {
queue.push(node -> left);
}
if(node -> right) {
queue.push(node -> right);
}
}
result.push_back(path);
}
return result;
}
};
结语
弄明白层序遍历的逻辑和代码后,就可以一举攻破很多道题目,《代码随想录》已经整理好了十道题目,之后的笔记我会一道一道将它们整理出来。