题目大意
#
给定一个二叉树,返回其「自底向上」,且按「层序遍历」得到的节点值。
解题思路
#
先得到层次遍历的节点顺序,再将其进行反转返回即可。
其中层次遍历用到了广度优先搜索,不过需要增加一些变化。普通广度优先搜索只取一个元素,变化后的广度优先搜索每次取出第 i 层上所有元素。
具体步骤如下:
- 根节点入队
- 当队列不为空时,求出当前队列长度 $s_i$
- 依次从队列中取出这 $s_i$ 个元素,将其左右子节点入队,然后继续迭代
- 当队列为空时,结束
代码
#
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
class Solution:
def levelOrderBottom(self, root: TreeNode) -> List[List[int]]:
if not root:
return []
queue = [root]
order = []
while queue:
level = []
size = len(queue)
for _ in range(size):
curr = queue.pop(0)
level.append(curr.val)
if curr.left:
queue.append(curr.left)
if curr.right:
queue.append(curr.right)
if level:
order.append(level)
return order[::-1]
|