BFS在二叉树层序遍历,最短路径问题的应用【Python】

灵感来源 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:
			
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值