算法模板-广度优先遍历

最新推荐文章于 2022-06-02 11:23:10 发布

maplezys

最新推荐文章于 2022-06-02 11:23:10 发布

阅读量195

点赞数 1

分类专栏：算法文章标签：算法宽度优先深度优先

本文链接：https://blog.csdn.net/qq_41006629/article/details/123071123

版权

算法专栏收录该内容

22 篇文章 0 订阅

订阅专栏

这篇博客介绍了广度优先遍历（BFS）的基本原理及其在解决实际问题中的应用。通过两个实例展示了BFS如何用于计算机器人能到达的格子数量以及按之字形顺序打印二叉树。代码实现清晰，强调了队列在BFS中的关键作用。

摘要由CSDN通过智能技术生成

简介

广度优先遍历（Breadth-First-Search，BFS），其特点便体现在广度，它是按“层“”来处理顶点，距离开始点最近的那些顶点首先被访问，而最远的那些顶点则最后被访问，这个和树的层序遍历很像。以下图为例：
在这里插入图片描述
BFS尽可能地搜索与同一个节点连接的所有节点，再进行下一次搜索。

队列实现

显然，DFS一般都与由队列实现：

def BFS(node):
	q = collections.deque([root])
	while q:
		temp = q.popleft()
		print(temp.val)
		q.append(temp.nexts) # 所有与temp相连的节点均入队列

练习题1

地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动，它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。例如，当k为18时，机器人能够进入方格 [35, 37] ，因为3+5+3+7=18。但它不能进入方格 [35, 38]，因为3+5+3+8=19。请问该机器人能够到达多少个格子？
处。
示例1：

输入：m = 2, n = 3, k = 1
输出：3

示例2：

输入：m = 3, n = 1, k = 0
输出：1

这题需要判断机器人能走到的区域内的格子数量，代码如下：

class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        q = collections.deque([(0, 0)])
        visited = set()
        count = 0
        while q:
            loc1, loc2 = q.popleft()           
            if loc1 < m and loc2 < n and (loc1, loc2) not in visited:
                visited.add((loc1, loc2))
                loc_k = sum([int(i) for i in str(loc1) + str(loc2)])
                if loc_k <= k:
                    count += 1
                    q.append((loc1 + 1, loc2))
                    q.append((loc1, loc2 + 1))
        return count

练习题2

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
示例：

给定二叉树：
    3
   / \
  9  20
    /  \
   15   7
 输出为：
 [
  [3],
  [20,9],
  [15,7]
]

这题就是树的层序遍历，在遍历时顺便记录层数，奇数层和偶数层输出顺序不同，代码如下：

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root: return []
        q, ans, flag = collections.deque(), [], 1
        q.append(root)
        while q:
            cur = collections.deque()
            while q:
                cur.append(q.popleft())
            ans_cur = []
            while cur:
                temp = cur.popleft()
                ans_cur.append(temp.val)
                if temp.left: q.append(temp.left)
                if temp.right: q.append(temp.right)
            ans.append(ans_cur if flag % 2 else ans_cur[::-1])
            flag += 1
        return ans