跳到主要内容

102. 二叉树的层序遍历 [medium]

102. 二叉树的层序遍历 [medium]

https://leetcode-cn.com/problems/binary-tree-level-order-traversal/

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例:

二叉树:[3,9,20,null,null,15,7],

    3
/ \
9 20
/ \
15 7

返回其层次遍历结果:

[
[3],
[9,20],
[15,7]
]

通过次数149,432 | 提交次数237,295

First Try

2020-06-24

算是比较快就完成了,只要记住宽度遍历BSF是通过队列实现的,其他level标记之类就比较顺理成章了。

# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None

class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
if root is None:
return []

queue = [(0, root)]
rv = []
while queue:
level, ele = queue.pop(0)
if len(rv) < level + 1:
rv.append([])
rv[level].append(ele.val)
if ele.left:
queue.append((level + 1, ele.left))
if ele.right:
queue.append((level + 1, ele.right))
return rv
  • 执行用时:12 ms, 在所有 Python 提交中击败了99.24%的用户
  • 内存消耗:13.3 MB, 在所有 Python 提交中击败了12.50%的用户