图的深度优先算法以及广度优先算法

最新推荐文章于 2024-05-09 11:45:55 发布

木易三水良

最新推荐文章于 2024-05-09 11:45:55 发布

阅读量104

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/qq_40722604/article/details/113121558

版权

算法专栏收录该内容

9 篇文章 0 订阅

订阅专栏

def dfs(isConnected):
    # 获取当前节点的邻居节点
    def _get_first_neighbor(isConnected, cur_point):
        j = len(isConnected) - 1
        neighbor = []
        while j >= 0:
            if isConnected[cur_point][j] == 1 and cur_point != j:
                neighbor.append(j)
            j -= 1
        return neighbor

    result = ''
    # 存储邻居节点
    neighbor_stack = []
    # 存储节点的访问状态，默认都是未被访问状态
    is_visited = [False] * len(isConnected)
    for cur_point in range(len(isConnected)):
        # 如果当前节点已经被访问，则对下一个节点进行访问
        if is_visited[cur_point]:
            continue
        # 当前节点置为已被访问状态
        is_visited[cur_point] = True
        result = result + str(cur_point) + '------>'
        # 当前节点的所有邻居节点压入栈中
        neighbor_stack.extend(_get_first_neighbor(isConnected, cur_point))
        # 如果邻居节点为空，说明当前已经是最后的一个节点，否则依次对灵雎节点进行处理
        while neighbor_stack.__len__() != 0:
            # 获取邻居节点
            pop = neighbor_stack.pop()
            # 如果邻居节点已经被访问，则继续处理下一个邻居节点
            if is_visited[pop]:
                continue
            else:
                # 对邻居节点进行遍历
                result = result + str(pop) + '------>'
                # 修改邻居节点的访问状态为已访问
                is_visited[pop] = True
                # 获取邻居节点的邻居节点
                neighbor_stack.extend(_get_first_neighbor(isConnected, pop))
    return result

def bfs(isConnected):
    # 获取当前节点的邻居节点
    def _get_first_neighbor(isConnected, cur_point):
        j = 0
        neighbor = []
        while j < len(isConnected):
            if isConnected[cur_point][j] == 1 and cur_point != j:
                neighbor.append(j)
            j += 1
        return neighbor

    result = ''
    # 存储邻居节点
    neighbor_stack = []
    # 存储节点的访问状态，默认都是未被访问状态
    is_visited = [False] * len(isConnected)
    for cur_point in range(len(isConnected)):
        # 如果当前节点已经被访问，则对下一个节点进行访问
        if is_visited[cur_point]:
            continue
        # 当前节点置为已被访问状态
        is_visited[cur_point] = True
        result = result + str(cur_point) + '------>'
        # 当前节点的所有邻居节点压入栈中
        neighbor_stack.extend(_get_first_neighbor(isConnected, cur_point))
        # 如果邻居节点为空，说明当前已经是最后的一个节点，否则依次对灵雎节点进行处理
        while neighbor_stack.__len__() != 0:
            # 获取邻居节点
            pop = neighbor_stack.pop(0)
            # 如果邻居节点已经被访问，则继续处理下一个邻居节点
            if is_visited[pop]:
                continue
            else:
                # 对邻居节点进行遍历
                result = result + str(pop) + '------>'
                # 修改邻居节点的访问状态为已访问
                is_visited[pop] = True
                # 获取邻居节点的邻居节点
                neighbor_stack.extend(_get_first_neighbor(isConnected, pop))
    return result

测试

isConnected = [
    # 1  2  3  4  5  6  7  8  9  10 11 12 13 14 15
    [1, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0],  # 1
    [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],  # 2
    [0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],  # 3
    [0, 0, 0, 1, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0],  # 4
    [0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0],  # 5
    [0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0],  # 6
    [0, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0],  # 7
    [1, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0],  # 8
    [0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 0],  # 9
    [0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 1],  # 10
    [0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0],  # 11
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0],  # 12
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0],  # 13
    [0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 1, 0],  # 14
    [0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1]  # 15
]
print("dfs结果：" + dfs(isConnected))
print("dfs结果：" + bfs(isConnected))