Leetcode102. 二叉树的层序遍历 Python3

Leetcode102. 二叉树的层序遍历

队列版层序遍历
python3代码：

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        # 首先判空
        if root is None:
            return []
        # 初始化队列，添加根节点
        que = [root]
        # 因为结果是二维的，先定义一个
        result = []
        # 队列不为空就不停循环
        while que:
            # tmp用来存储下一层的所有节点
            tmp = []
            # tmp_val用来存当前层的所有节点值，注意是节点的值，而且是从左到右
            tmp_val = []
            # 遍历队列中每个节点
            for node in que:
                # 将节点的值存入tmp_val
                tmp_val.append(node.val)
                # 将下一层的所有节点存入tmp
                if node.left:
                    tmp.append(node.left)
                if node.right:
                    tmp.append(node.right)
            # 将当前层的节点值列表存入result
            result.append(tmp_val)
            # 将下一层的所有节点列表替换现有的队列
            que = tmp
        # 返回结果
        return result