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%的用户