102. 二叉树的层序遍历

题目-中等难度

给你二叉树的根节点 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

1. bfs

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        # 初始化列表, 为树根节点
        li = [root]
        # 结果列表存储每一行节点值
        res = []
        # 遍历树每一行
        while li:
            # 列表存储当前行节点值
            resLi = []
            # 遍历当前行
            for _ in range(len(li)):
                # 获取当前行节点
                a = li.pop(0)
                # 当节点存在
                if a:
                    # 添加节点值到resLi列表
                    resLi.append(a.val)
                    # 判断是否存在左右子节点, 若存在, 添加到li, 作为下一行需要遍历的对象节点
                    if a.left:
                        li.append(a.left)
                    if a.right:
                        li.append(a.right)
            # 为空的情况不添加
            if resLi:
                res.append(resLi)
        return res