灵感来源 leetcode题解
由于上述题解都是基于java,而python的TreeNode不太一样,所以我这里重新组织一下。
如果我们使用 DFS/BFS 只是为了遍历一棵树、一张图上的所有结点的话,那么 DFS 和 BFS 的能力没什么差别,我们当然更倾向于更方便写、空间复杂度更低的 DFS 遍历。不过,某些使用场景是 DFS 做不到的,只能使用 BFS 遍历。这就是本文要介绍的两个场景:「层序遍历」、「最短路径」。
本文包括以下内容:
- DFS 与 BFS 的特点比较
- BFS 的适用场景
- 如何用 BFS 进行层序遍历
- 如何用 BFS 求解最短路径问题
我先用Python实现层序遍历的一维数组,正常来说只用在一个队列中添加节点即可,但是Python貌似不支持在list中添加节点,所以我额外用了一个values列表来存志节点的值。
def level(root):
if root is None:
return []
else:
queue, values= [], []
node = root
queue.append(node)
while True:
for _ in range(len(queue)): #只会执行一层的数目
node=queue.pop()
if node.left:
queue.append(node)
values.append(node.val)
if node.right:
queue.append(node)
values.append(node.val)
if len(queue)==0:
break
return values
上述代码返回的是一个近似层序遍历的一维数组,我们还需要把每一层区分开。很自然会想到用一个临时的列表来存储每一层。然后每个列表都添加到values,而不是添加每个元素。
def level_(root):
if root is None:
return []
else:
node,queue,values = root,[],[]
queue.append(node)
values.append(node.val)
while True: