六种图算法的python实现

Sun_Raiser

已于 2024-06-11 17:03:02 修改

阅读量953

点赞数 11

分类专栏： algorithm 文章标签：算法 python 开发语言

于 2024-06-11 16:49:37 首次发布

本文链接：https://blog.csdn.net/Sun_Raiser/article/details/139603214

版权

algorithm 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

六种图算法的python实现

1. Prim 算法

基本原理

Prim算法是一种求解最小生成树的贪心算法。所谓最小生成树，就是对于给定的连通图，找到一棵包含所有顶点的树，且树上所有边的权重之和最小。Prim算法从一个顶点开始，每次选择与当前生成树距离最短的顶点加入到生成树中，直到所有顶点都加入为止。

应用场景

Prim算法广泛应用于网络设计、电路设计、道路规划等领域，其中需要找到连接所有点的最小成本路径。例如，在铺设电缆或道路网络中，可以使用Prim算法来找到连接所有地点所需的最小成本路径。

优缺点

优点：

Prim算法能够找到全局最优解，即最小生成树。
适用于稠密图，即边数较多的图。

缺点：

对于稀疏图（边数较少的图），Prim算法可能不是最高效的选择。
Prim算法需要频繁地查找和更新距离，因此在实现上可能需要使用优先队列等数据结构来提高效率。

Python 实现

以下是一个简单的Prim算法的Python实现：

import heapq

def prim(graph):
    # 初始化
    start_node = 'A'  # 可以选择任意节点作为起始节点
    visited = set([start_node])  # 已访问节点的集合
    edges = [
        (cost, start_node, to)
        for to, cost in graph[start_node].items()
    ]  # 初始可选边集合
    heapq.heapify(edges)  # 使用堆来优化查找最小边的过程
    mst = []  # 最小生成树的边
    total_cost = 0  # 最小生成树的总权重

    while edges:
        cost, frm, to = heapq.heappop(edges)  # 弹出权重最小的边
        if to not in visited:  # 如果目标节点未被访问过
            visited.add(to)  # 将其标记为已访问
            mst.append((frm, to, cost))  # 将这条边加入最小生成树
            total_cost += cost  # 更新总权重
            # 将与新加入节点相连且未访问过的节点（及其边）加入堆中
            for to_next, cost2 in graph[to].items():
                if to_next not in visited:
                    heapq.heappush(edges, (cost2, to, to_next))

    return mst, total_cost

# 示例图（以字典形式表示，键为起点，值为一个字典，表示从起点到各个终点的权重）
graph = {
    'A': {'B': 2, 'C': 3},
    'B': {'A': 2, 'C': 1, 'D': 1, 'E': 4},
    'C': {'A': 3, 'B': 1, 'F': 5},
    'D': {'B': 1, 'E': 1},
    'E': {'B': 4, 'D': 1, 'F': 1},
    'F': {'C': 5, 'E': 1, 'G': 1},
    'G': {'F': 1}
}

mst, total_cost = prim(graph)
print("最小生成树的边:", mst)
print("最小生成树的总权重:", total_cost)

原始图：

在这里插入图片描述

最小生成树：

在这里插入图片描述

这个实现使用了优先队列（通过Python的heapq模块实现）来优化查找最小边的过程。注意，这个实现假设图是连通的，且权重都是正数。如果图不连通，或者存在负权重的边，那么这个实现可能不适用。

2. Kruskal 算法

基本原理

Kruskal算法是一种用于查找最小生成树的算法，属于贪心算法的一种。在一个加权无向图中，最小生成树是指一个边集，它能够连接所有的顶点，并且所有边的权值之和最小，而且不包含环。

Kruskal算法的基本思想是：

将图中的所有边按照权重从小到大排序。
从最小的边开始，依次选择边，每次选择一条边后，判断这条边是否会与已选择的边形成环。
如果不会形成环，则将这条边加入到最小生成树中；否则，放弃这条边。
重复步骤2和3，直到所有的顶点都连接在一起或者所有的边都已被考虑过。

应用场景

Kruskal算法在多个领域有广泛的应用，包括但不限于：

网络设计：在构建计算机网络、电力网络或交通网络时，需要找到连接所有节点的最小成本树。
电路设计：在电子设计中，需要找到连接所有组件的最短电线长度。

优缺点

优点：

算法简单易懂，容易实现。
总是能找到全局最优解，即权重和最小的生成树。

缺点：

当图中的边数非常多时，排序操作可能会比较耗时。
需要额外的数据结构（如并查集）来检测环的形成，增加了算法的复杂度。

Python 实现

下面是一个简单的Kruskal算法的Python实现，使用了并查集来检测环：

class UnionFind:
    def __init__(self, n):
        self.parent = list(range(n))
    
    def find(self, x):
        if self.parent[x] != x:
            self.parent[x] = self.find(self.parent[x])
        return self.parent[x]
    
    def union(self, x, y):
        root_x = self.find(x)
        root_y = self.find(y)
        self.parent[root_x] = root_y

def kruskal(graph):
    edges = []
    for u in graph:
        for v, w in graph[u]:
            edges.append((w, u, v))
    edges.sort()  # 按权重排序
    
    uf = UnionFind(len(graph))
    mst = []
    for w, u, v in edges:
        if uf.find(u) != uf.find(v):
            uf.union(u, v)
            mst.append((u, v, w))
    return mst

# 示例图
graph = {
    0: [(1, 7), (2, 9), (5, 14)],
    1: [(0, 7), (2, 10), (3, 15)],
    2: [(0, 9), (1, 10), (3, 11), (5, 2)],
    3: [(1, 15), (2, 11), (4, 6)],
    4: [(3, 6), (5, 9)],
    5: [(0, 14), (2, 2), (4, 9)]
}

mst = kruskal(graph)
print("最小生成树的边：", mst)

原始图：

在这里插入图片描述

最小生成树：

在这里插入图片描述

这个代码示例中，graph是一个字典，表示图的邻接表形式。kruskal函数首先将所有边按照权重排序，然后使用并查集来检测并避免环的形成，最后返回最小生成树的边列表。

3. Floyd-Warshall 算法

基本原理

Floyd-Warshall 算法是一种用于寻找给定加权图中所有顶点对之间最短路径的算法。它适用于带权重的有向图或带权重的无向图，并且可以处理负权重的边，但不能处理含有负权重环的图。

算法的基本思想是通过逐步构建更长的路径来找到最短路径。它使用一个三维数组 dist[k][i][j] 来表示从顶点 i 到顶点 j 的、所有中间顶点来自集合 {0, 1, …, k} 的一条最短路径的权重。最终，当 k 等于图中顶点数量减一时，dist[k][i][j] 就是从 i 到 j 的最短路径长度。

在实际应用中，为了节省空间，通常会将三维数组简化为二维数组，通过逐步更新这个二维数组来记录最短路径。

应用场景

Floyd-Warshall 算法通常用于需要找到图中所有顶点对之间最短路径的场景，例如网络路由选择、社交网络中的最短关系链查找等。

优缺点

优点：

算法简单易懂，实现起来较为方便。
能够处理带有负权重的边。
可以一次性计算出所有顶点对之间的最短路径。

缺点：

时间复杂度和空间复杂度都是 O(n^3)，其中 n 是图中的顶点数。对于大型图来说，这可能是一个问题。
不能处理带有负权重环的图。

Python 实现

以下是一个简单的 Floyd-Warshall 算法的 Python 实现：

def floyd_warshall(graph):
    n = len(graph)
    dist = [[float('inf')] * n for _ in range(n)]

    # 直接从邻接矩阵复制权重到dist矩阵
    for i in range(n):
        for j in range(n):
            dist[i][j] = graph[i][j]
            # 设置对角线上的距离为0，表示节点到自身的距离为0
            if i == j:
                dist[i][j] = 0

    # Floyd-Warshall算法的核心部分
    for k in range(n):
        for i in range(n):
            for j in range(n):
                if dist[i][k] != float('inf') and dist[k][j] != float('inf'):
                    dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j])
    return dist


# 示例图，使用邻接矩阵表示  
graph = [
    [0, 3, float('inf'), 5],
    [2, 0, float('inf'), 4],
    [float('inf'), 1, 0, float('inf')],
    [float('inf'), float('inf'), 2, 0]
]
print(floyd_warshall(graph))

这个实现中，graph 是一个邻接矩阵，表示图中的边和权重。float('inf') 表示无穷大，用于表示两个顶点之间没有直接连接的情况。

4. Bellman-Ford 算法

基本原理

Bellman-Ford 算法是一种用于在带权图中找出从单一源点到其他所有点的最短路径的算法。它适用于带有负权边的图，这是它与 Dijkstra 算法的一个重要区别。Bellman-Ford 算法的基本思想是对图中的所有边进行松弛操作，逐步逼近最短路径。

算法步骤如下：

初始化：设置源点到所有其他点的距离为无穷大（或一个非常大的数），源点到自身的距离为 0。
松弛操作：对所有边进行多次遍历（边的数量减 1 次），每次遍历时，对于图中的每一条边 (u, v)，如果当前源点到点 u 的距离加上边 (u, v) 的权重小于源点到点 v 的当前距离，则更新源点到点 v 的距离为更小的值。
检查负权环：再进行一次边的遍历，如果仍有距离被更新，则说明图中存在负权环，因为在一个没有负权环的图中，最短路径的长度在经过 |V|-1 次松弛操作后就已经确定，其中 |V| 是顶点的数量。

应用场景

Bellman-Ford 算法适用于需要处理带有负权重的图的最短路径问题。例如，在路由选择、网络流优化、交通运输规划等领域中，边的权重可能表示时间、费用或距离，并且可能为负（例如，表示某种形式的“奖励”或“折扣”）。

优缺点

优点：

可以处理带有负权重的图。
可以检测出负权环。
算法实现相对简单。

缺点：

时间复杂度较高，为 O(|V|*|E|)，其中 |V| 是顶点数，|E| 是边数。对于稠密图来说，这可能是一个问题。
在没有负权重的图中，通常不如 Dijkstra 算法高效。

Python 实现

以下是一个简单的 Bellman-Ford 算法的 Python 实现：

class Graph:
    def __init__(self, vertices):
        self.V = vertices
        self.graph = []

    def add_edge(self, u, v, w):
        self.graph.append([u, v, w])

    def bellman_ford(self, src):
        dist = [float('inf')] * self.V
        dist[src] = 0

        # 松弛操作，进行 |V|-1 次
        for _ in range(self.V - 1):
            for u, v, w in self.graph:
                if dist[u] != float('inf') and dist[u] + w < dist[v]:
                    dist[v] = dist[u] + w

        # 检查负权环
        for u, v, w in self.graph:
            if dist[u] != float('inf') and dist[u] + w < dist[v]:
                return "图中存在负权环"

        return dist

# 使用示例
g = Graph(5)
g.add_edge(0, 1, -1)
g.add_edge(0, 2, 4)
g.add_edge(1, 2, 3)
g.add_edge(1, 3, 2)
g.add_edge(1, 4, 2)
g.add_edge(3, 2, 5)
g.add_edge(3, 1, 1)
g.add_edge(4, 3, -3)

print(g.bellman_ford(0))  # 输出从顶点 0 到其他各顶点的最短距离

这个实现创建了一个简单的图数据结构，并实现了 Bellman-Ford 算法来查找从指定源点到所有其他点的最短路径。注意，如果图中存在负权环，算法会返回相应的提示信息。

5. 强连通分量算法

基本原理

强连通分量算法主要用于在有向图中找出强连通分量。强连通分量是指在有向图中，任意两个顶点之间都存在一条路径，即它们是相互可达的。常见的强连通分量算法有Kosaraju算法和Tarjan算法。

这里，我们主要介绍Kosaraju算法，其基本原理是通过两次深度优先搜索（DFS）来找到强连通分量。

第一次DFS：对原始图进行深度优先搜索，并记录每个节点的完成时间（即DFS遍历结束时的时间）。
计算转置图：将原始图的所有边反向，得到转置图。
第二次DFS：根据第一次DFS记录的节点完成时间进行降序排序，然后按照这个顺序对转置图进行DFS。每次DFS遍历得到的子图就是一个强连通分量。

应用场景

强连通分量算法常用于社交网络分析、网页链接分析、程序控制流图分析等领域，用于找出紧密相关的群体或模块。

优缺点

优点：

能够准确地找出有向图中的强连通分量。
Kosaraju算法相对直观，易于理解。

缺点：

需要进行两次DFS，时间复杂度相对较高。
需要额外的空间来存储转置图和节点的完成时间。

Python实现（Kosaraju算法）

from collections import defaultdict


class Graph:
    def __init__(self, vertices):
        self.V = vertices
        self.graph = defaultdict(list)

    def addEdge(self, u, v):
        self.graph[u].append(v)

    def DFSUtil(self, v, visited):
        visited.add(v)
        print(v, end='')
        for i in self.graph[v]:
            if i not in visited:
                self.DFSUtil(i, visited)

    def fillOrder(self, v, visited, stack):
        visited.add(v)
        for i in self.graph[v]:
            if i not in visited:
                self.fillOrder(i, visited, stack)
        stack.append(v)  # 修正这里，不要重新赋值给 stack

    def getTranspose(self):
        g = Graph(self.V)
        for i in self.graph:
            for j in self.graph[i]:
                g.addEdge(j, i)
        return g

    def printSCCs(self):
        stack = []
        visited = set()
        for i in range(self.V):
            if i not in visited:
                self.fillOrder(i, visited, stack)
        gr = self.getTranspose()
        visited = set()
        while stack:
            i = stack.pop()
            if i not in visited:
                gr.DFSUtil(i, visited)
                print()


# 使用示例
g = Graph(5)
g.addEdge(1, 0)
g.addEdge(0, 2)
g.addEdge(2, 1)
g.addEdge(0, 3)
g.addEdge(3, 4)
print("强连通分量为:")
g.printSCCs()

注意：上述代码是一个简化的示例，用于说明Kosaraju算法的基本原理。在实际应用中，可能需要根据具体需求进行适当的修改和优化。

这个Python实现中，Graph类表示一个有向图，addEdge方法用于添加边，DFSUtil方法用于执行深度优先搜索，fillOrder方法用于第一次DFS并记录节点的完成时间（通过栈的顺序隐式表示），getTranspose方法用于计算转置图，printSCCs方法用于找出并打印所有的强连通分量。

6. 拓扑排序算法

基本原理

拓扑排序（Topological Sorting）是针对有向无环图（DAG, Directed Acyclic Graph）的顶点的一种排序，它将有向无环图的顶点以线性序列的方式输出，对于每一条有向边 (u, v)，顶点 u 在排序中都出现在顶点 v 之前。拓扑排序常用于任务调度、制定课程学习计划等场景。

拓扑排序的基本原理是通过不断移除入度为0的节点（即没有前驱或依赖的节点），并更新其相邻节点的入度，直到所有节点都被移除或者无法再移除入度为0的节点为止。如果还有剩余节点，则说明图中存在环，无法进行拓扑排序。

应用场景

任务调度：在项目管理中，常常需要确定任务的执行顺序，拓扑排序可以帮助确定哪些任务可以并行执行，哪些任务必须等待其他任务完成后才能开始。
制定学习计划：在学习多门有依赖关系的课程时，拓扑排序可以帮助规划学习顺序。
编译器中的指令重排：在编译器优化中，拓扑排序可以确保指令以正确的顺序执行，从而提高程序的执行效率。

优缺点

优点：

可以有效地确定有依赖关系的任务或操作的执行顺序。
算法相对简单，容易实现。

缺点：

仅适用于有向无环图（DAG），如果图中存在环，则算法无法得出结果。
拓扑排序的结果不唯一，因为可能存在多个入度为0的节点，这些节点的处理顺序可以不同。

Python实现

以下是一个简单的拓扑排序的Python实现：

from collections import defaultdict, deque


def topological_sort(graph):
    # 统计入度
    indegree = defaultdict(int)
    for node in graph:
        for neighbor in graph[node]:
            indegree[neighbor] += 1

    # 将入度为0的节点加入队列
    queue = deque([node for node in graph if indegree[node] == 0])
    result = []

    # 拓扑排序
    while queue:
        node = queue.popleft()
        result.append(node)
        # 将该节点的邻接节点的入度减1，并将入度为0的节点加入队列
        for neighbor in graph[node]:
            indegree[neighbor] -= 1
            if indegree[neighbor] == 0:
                queue.append(neighbor)

    # 如果结果中的节点数等于图中的节点数，则拓扑排序成功
    if len(result) == len(graph):
        return result
    else:
        return None


# 测试
graph = {
    'A': ['B', 'C'],
    'B': ['D', 'E'],
    'C': ['F'],
    'D': [],
    'E': ['F'],
    'F': []
}

sorted_nodes = topological_sort(graph)
if sorted_nodes:
    print("拓扑排序结果:", sorted_nodes)
else:
    print("图中存在环，无法进行拓扑排序")

这个示例中，graph是一个字典，表示有向图的结构，其中键是节点，值是该节点的相邻节点列表。topological_sort函数接受这个图结构作为输入，并返回一个可能的拓扑排序结果列表。如果图中存在环，则返回相应的错误信息。

Sun_Raiser

关注

11
点赞
踩
12

收藏

觉得还不错? 一键收藏
打赏
2
评论
六种图算法的python实现

Prim算法是一种求解最小生成树的贪心算法。所谓最小生成树，就是对于给定的连通图，找到一棵包含所有顶点的树，且树上所有边的权重之和最小。Prim算法从一个顶点开始，每次选择与当前生成树距离最短的顶点加入到生成树中，直到所有顶点都加入为止。
复制链接

扫一扫