Description
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
分析
题目的意思是:这道题要用队列来实现,以前用C++实现过一次,用python还写了几个bug,哈哈哈,看来不复习迟早会出事。
代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if(root is None):
return []
q = deque()
q.append(root)
res=[]
while(q):
t=[]
for _ in range(len(q)):
p=q.popleft()
t.append(p.val)
if(p.left):
q.append(p.left)
if(p.right):
q.append(p.right)
res.append(t)
return res