代码随想录第15天|二叉树的层序遍历

二叉树的层序遍历

102. 二叉树的层序遍历 - 力扣（LeetCode）

代码随想录 (programmercarl.com)

讲透二叉树的层序遍历 | 广度优先搜索 | LeetCode：102.二叉树的层序遍历_哔哩哔哩_bilibili

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

示例 2：

输入：root = [1]
输出：[[1]]

示例 3：

输入：root = []
输出：[]

提示：

• 树中节点数目在范围 [0, 2000] 内
• -1000 <= Node.val <= 1000

一开始对这个题目没什么思路。看了卡哥题解：

大致思路就是用一个队列来遍历，因为队列满足先进先出的原则，按照这个题目，需要返回的是一个集合，集合里边放的是数组，每个数组容纳了每一层的数字。

首先定义一个最终的集合，用来存储数组：

class Solution{
    public List<List<Integer>> resList = new ArrayList<List<List<Integer>>;
    
        
}

创建一个队列，用于存储待遍历的结点：

Queue<TreeNode> que = new LinkedList<TreeNode>();
que.offer(node);

生成一个辅助方法，用于层序遍历：
 

public void checkFund(TreeNode node){
    if(node == null){//如果结点为空，则返回空
        return null;
    }
}

循环遍历，将二叉树中的结点加到集合中，知道二叉树为空：

while(!que.isEmpty()){
    
}

创建一个列表，用于存储当前层的值：

List<Integer> itemList = new ArrayList<Integer>();
int length = que.size();

循环直到len=0时，停止遍历。len=0时，说明待遍历的结点数为0.

while(len>0){
    
}

从队列中取出一个结点并放入临时节点中：（放入临时节点主要是为了在把节点弹出之前还能放入该节点的左右节点）

TreeNode tmpNode = que.poll();

将当前节点的值放到列表中：

itemList.add(tmpNode.val);

如果当前节点的左右孩子节点不为空，则将左右孩子节点放入队列中：
 

if(tmpNode.left != null) que.offer(tmpNode.left);
if(tmpNode.right != null) que.offer(tmpNode.right);
len--;

将当前层的列表添加到结果列表中：

resList.add(itemList);

最终代码：

// 102.二叉树的层序遍历
class Solution {
    // 存储层序遍历结果的列表
    public List<List<Integer>> resList = new ArrayList<List<Integer>>();

    // 层序遍历方法，接收树的根节点作为参数
    public List<List<Integer>> levelOrder(TreeNode root) {
        // 调用层序遍历的方法
        checkFun02(root);

        // 返回层序遍历结果列表
        return resList;
    }

    // BFS--迭代方式--借助队列
    // 辅助方法，进行层序遍历
    public void checkFun02(TreeNode node) {
        // 如果节点为空，直接返回
        if (node == null) return;

        // 创建一个队列，用于存储待遍历的节点
        Queue<TreeNode> que = new LinkedList<TreeNode>();
        // 将根节点加入队列
        que.offer(node);

        // 循环直到队列为空
        while (!que.isEmpty()) {
            // 创建一个列表，用于存储当前层的节点值
            List<Integer> itemList = new ArrayList<Integer>();
            // 获取当前队列的长度，即当前层的节点个数
            int len = que.size();

            // 循环直到当前层的节点全部处理完
            while (len > 0) {
                // 从队列中取出一个节点
                TreeNode tmpNode = que.poll();
                // 将当前节点的值添加到当前层的列表中
                itemList.add(tmpNode.val);

                // 如果当前节点的左子节点不为空，则将其加入队列
                if (tmpNode.left != null) que.offer(tmpNode.left);
                // 如果当前节点的右子节点不为空，则将其加入队列
                if (tmpNode.right != null) que.offer(tmpNode.right);

                // 当前层的节点数减一
                len--;
            }

            // 将当前层的列表添加到结果列表中
            resList.add(itemList);
        }
    }
}