基于二维矩阵的深搜（dfs）和广搜（bfs）python实现

最新推荐文章于 2024-08-09 21:26:42 发布

AivenZ

最新推荐文章于 2024-08-09 21:26:42 发布

阅读量2.2k

点赞数 1

分类专栏：算法与数据结构

本文链接：https://blog.csdn.net/AivenZhong/article/details/85009225

版权

本文介绍了如何在非邻接矩阵的二维结构中，使用Python实现深度优先搜索（DFS）和广度优先搜索（BFS）。每个节点仅与其上、下、左、右四个相邻节点相连。

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

该二维矩阵不是邻接矩阵，每个结点只和上下左右4个方向的结点有连接。

def bfs(start, matrix):
    """广搜"""
    n = len(matrix)
    queue = []
    queue.append(start)
    vis = [[False for _ in range(n)] for _ in range(n)]
    vis[start[0]][start[1]] = True
    dires = [[0, 1], [0, -1], [1, 0], [-1, 0]]
    while queue:
        x, y = queue.pop(0)
        print(matrix[x][y], end=' ')
        for dx, dy in dires:
            nx, ny = x + dx, y + dy
            if 0 <= nx < n and 0 <= ny < n and not vis[nx][ny]:
                queue.append([nx, ny])
                vis[nx][ny] = True


def dfs(start, matrix, vis):
    """深搜"""
    dires = [[0, 1], [0, -1], [1, 0], [-1, 0]]
    x, y = start[0], start[1]
    print(matrix[x][y], end=' ')
    vis[x][y] = True
    for dx, dy in dires:
        nx, ny = x + dx, y + dy
        if 0 <= nx < n