forked from Teingi/python-test
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathleetcode_107.py
33 lines (30 loc) · 1.25 KB
/
leetcode_107.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#leetcode 107. 二叉树的层次遍历 II
# 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 levelOrderBottom(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
#bfs广度优先搜索,不熟悉可以网上参考bfs代码
if not root:
return []#为空则返回空列表
queue=[root]#使用列表实现队列的功能,首先存储root
res=[]
while queue:#当queue不为空时
nodes=[]#存节点,每次循环前置空,每次只装一部分
node_values=[]#存节点的值
for node in queue:
if node.left:
nodes.append(node.left)#将左子树装入队列中
if node.right:
nodes.append(node.right)
node_values+=[node.val]#因为每次循环node_values都会置空,所以最终结果保存在res里,node_values只是一小部分结果
res=[node_values]+res#实现从底到顶,node_values放前面.
queue=nodes#将新添加的节点重新赋值给queue
return res