zlatan的算法笔记13--二叉树的层序遍历（力扣102题）

算法笔记

之前谈到了二叉树的前中序遍历，也给出了非递归与递归两种情况下的代码，本次算法笔记将继续学习一种遍历方法，层序遍历。层序遍历可以理解为广度优先遍历，即从左到右遍历每一层节点即可。

题目描述

解题过程

对于层序遍历，我们需要使用数据结构队列去实现，因为层序遍历的过程符合队列先进先出的特点。拿题目描述中的例子举例，先将根节点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;
    }
};

结语

弄明白层序遍历的逻辑和代码后，就可以一举攻破很多道题目，《代码随想录》已经整理好了十道题目，之后的笔记我会一道一道将它们整理出来。