102. 二叉树的层序遍历 (Python 实现)

x33g5p2x  于2022-02-18 转载在 Python  
字(0.6k)|赞(0)|评价(0)|浏览(166)

题目:
给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

示例 1:

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

示例 2:

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

示例 3:

输入:root = []
输出:[]

代码:

# 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: TreeNode) -> List[List[int]]:
        if not root:
            return []
        queue, res = [], []
        queue.append(root)
        while queue:
            k = []
            for _ in range(len(queue)):
                temp = queue.pop(0)
                k.append(temp.val)
                if temp.left:
                    queue.append(temp.left)
                if temp.right:
                    queue.append(temp.right)
            res.append(k)
        return res

相关文章