102. 二叉树的层序遍历 Python

---

一、题目描述

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

二、代码

代码如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
        if root is None:
            return []
        result = []
        # 最开始的一层中只有根节点一个节点
        cur = [root]
        while cur:
            next = []
            path = []
            for node in cur:
                # path 用于存储这一层所有的结点值
                path.append(node.val)
                # 如果当前节点有左子树
                if node.left:  
                    # 则在下一层需要遍历的结点数组中添加
                    next.append(node.left)
                # 如果当前节点有右子树
                if node.right: 
                    next.append(node.right)
            # 让当前遍历的结点数组更新为下一次遍历的结点数组，即下一层所有结点的数组
            cur = next
            result.append(path)
        return result

---

三、解题思路

本题考察二叉树的层序遍历，且需要按照每层的结点值返回结果数组。
本题解采用while循环遍历二叉树，用cur数组来存储当前层的所有结点，每次遍历结束后添加当前层的结点值到result中，然后更新cur数组，代表下一层需要遍历的结点有什么（用next数组存储），当cur中没有元素时则代表遍历到二叉树最底层，遍历结束，返回result即可。