Python中实现图论算法

简介: Python中实现图论算法 “【5月更文挑战第20天】”

在Python中实现图论算法,我们通常会用到networkx这个库,它是一个强大的图论库,提供了丰富的数据结构和算法。以下是一个使用networkx库实现的图的深度优先搜索(DFS)算法的例子:

首先,确保你已经安装了networkx库。如果没有安装,可以通过以下命令安装:

pip install networkx

然后,我们可以编写一个简单的程序来实现DFS:

import networkx as nx

# 创建一个无向图
G = nx.Graph()

# 添加节点和边
G.add_edge(1, 2)
G.add_edge(1, 3)
G.add_edge(2, 4)
G.add_edge(3, 4)
G.add_edge(4, 5)

# 打印图的边
print("Edges in graph:", G.edges())

# 定义DFS函数
def dfs(graph, start):
    visited, stack = set(), [start]
    while stack:
        vertex = stack.pop()
        if vertex not in visited:
            visited.add(vertex)
            stack.extend(graph.successors(vertex))
            print(vertex, end=' ')
    return visited

# 调用DFS函数并打印结果
print("\nDFS starting from node 1:")
dfs(G, 1)

在这个例子中,我们首先创建了一个无向图G,然后添加了一些边。接着,我们定义了一个dfs函数,它接受图和一个起始节点作为参数,执行深度优先搜索,并打印出遍历的节点顺序。

请注意,networkx本身也提供了DFS的实现,可以通过networkx.dfs_preorder_nodesnetworkx.dfs_postorder_nodes等函数直接使用。

此外,如果你想要实现自己的数据结构和算法而不是依赖networkx,下面是一个不使用networkx的DFS实现示例:

# 使用字典来表示图
graph = {
   
    1: [2, 3],
    2: [1, 4],
    3: [1, 4],
    4: [2, 3, 5],
    5: [4]
}

# 辅助函数:DFS的递归实现
def dfs_recursive(graph, node, visited):
    if node in visited:
        return
    print(node, end=' ')
    visited.add(node)
    for neighbour in graph[node]:
        dfs_recursive(graph, neighbour, visited)

# 调用DFS递归函数并传入起始节点1
visited = set()
dfs_recursive(graph, 1, visited)

在这个例子中,我们使用字典来表示图,其中键是节点,值是与该节点相连的节点列表。dfs_recursive函数是一个递归实现的DFS,它接受图、当前节点和已访问节点集合作为参数。

目录
相关文章
|
4月前
|
算法 搜索推荐 JavaScript
基于python智能推荐算法的全屋定制系统
本研究聚焦基于智能推荐算法的全屋定制平台网站设计,旨在解决消费者在个性化定制中面临的选择难题。通过整合Django、Vue、Python与MySQL等技术,构建集家装设计、材料推荐、家具搭配于一体的一站式智能服务平台,提升用户体验与行业数字化水平。
|
4月前
|
存储 监控 算法
监控电脑屏幕的帧数据检索 Python 语言算法
针对监控电脑屏幕场景,本文提出基于哈希表的帧数据高效检索方案。利用时间戳作键,实现O(1)级查询与去重,结合链式地址法支持多条件检索,并通过Python实现插入、查询、删除操作。测试表明,相较传统列表,检索速度提升80%以上,存储减少15%,具备高实时性与可扩展性,适用于大规模屏幕监控系统。
178 5
|
5月前
|
存储 算法 调度
【复现】【遗传算法】考虑储能和可再生能源消纳责任制的售电公司购售电策略(Python代码实现)
【复现】【遗传算法】考虑储能和可再生能源消纳责任制的售电公司购售电策略(Python代码实现)
278 26
|
5月前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于D*算法的机器人路径规划(Python代码实现)
【机器人路径规划】基于D*算法的机器人路径规划(Python代码实现)
304 0
|
5月前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于改进型A*算法的机器人路径规划(Python代码实现)
【机器人路径规划】基于改进型A*算法的机器人路径规划(Python代码实现)
430 0
|
5月前
|
机器学习/深度学习 编解码 算法
【机器人路径规划】基于迪杰斯特拉算法(Dijkstra)的机器人路径规划(Python代码实现)
【机器人路径规划】基于迪杰斯特拉算法(Dijkstra)的机器人路径规划(Python代码实现)
507 4
|
5月前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于A*算法的机器人路径规划研究(Python代码实现)
【机器人路径规划】基于A*算法的机器人路径规划研究(Python代码实现)
720 4
|
5月前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于深度优先搜索(Depth-First-Search,DFS)算法的机器人路径规划(Python代码实现)
【机器人路径规划】基于深度优先搜索(Depth-First-Search,DFS)算法的机器人路径规划(Python代码实现)
319 3
|
5月前
|
算法 机器人 定位技术
【机器人路径规划】基于流场寻路算法(Flow Field Pathfinding)的机器人路径规划(Python代码实现)
【机器人路径规划】基于流场寻路算法(Flow Field Pathfinding)的机器人路径规划(Python代码实现)
315 4
机器学习/深度学习 算法 自动驾驶
1057 0

推荐镜像

更多