- 条件最短路径问题,指带有约束条件、限制条件的最短路径问题。例如: 顶点约束,包括必经点或禁止点的限制; 边的约束,包括必经路段、禁行路段和单向路段;无权路径长度的限制,如要求经过几步或不超过几步到达终点。
- 本文基于 NetworkX 工具包,建立了一个遍历简单路径、判断约束条件的通用框架。
- 数模竞赛真题案例,详解禁止点、禁止边、必经点、必经边的约束条件处理,进而可以扩展到任何约束条件。
- 『Python小白的数学建模课 @ Youcans』带你从数模小白成为国赛达人。
1. 带有条件约束的最短路径问题
最短路径问题是图论中求两个顶点之间的最短路径问题,通常是求最短加权路径。
在数模竞赛中,一般不会直接求最短路径,那就太简单了,总是要在基本问题中设置各种限制条件,情况就完全不同了。
条件最短路径问题,指带有约束条件、限制条件的最短路径问题。例如: 顶点约束,包括必经点或禁止点的限制; 边的约束,包括必经路段、禁行路段和单向路段;无权路径长度的限制,如要求经过几步或不超过几步到达终点。
进一步地,还有双目标限制的最短路径问题,求最短长度中花费最小的路线;交通限制条件下的最短路径问题,如转向限制、道路交叉口的时间延误的约束。
求解带有限制条件的最短路径问题,总体来说可以分为两类基本方法: 一类是基于不带限制条件的最短路径算法,对求解过程中的每一条有效路径,都用限制条件进行判断,如果满足所有限制条件则继续,如果不满足限制条件则放弃该路径; 另一类方法是基于具体问题和选择算法的特点,将问题转化为有约束的规划问题来处理。
但是,如果使用 NetworkX 求解带有限制条件的最短路径问题,采用这两类方法都会有一些困难。原因在于前文所介绍的 NetworkX 提供的 Dijkstra 算法、Bellman-Ford 算法、Floyd 算法和启发式算法 A* 都是封装函数,没有提供设置约束条件的选项和接口,因此用户不能把条件判断语句加入这些封装函数的程序内部。
这个问题不仅存在于 Python 语言的 NetworkX 工具包,对于其它计算机语言的工具包也是类似的。自己编程序费时费力,但可以根据需要修改和扩展;直接调用工具包的算法函数非常方便,但不能进行修改或扩展。
不过,NetworkX 可以生成两个顶点之间的所有简单路径,而且可以获得所有简单路径的边的列表。利用简单路径算法,可以通过对约束条件的判断来求解带有顶点约束和边约束的最短路径问题。
2. 案例:蚂蚁的最优路径分析
2.1 问题描述
蚁巢有若干个储藏间(图中圆圈所示),储藏间之间有路径相连(路径拓扑结构如图所示)。
该图为无向图,路径通行的花费如图中线路上的数字所示,路径正反方向通行的花费相同。
要求从起点 N0 到终点 N17 的最优路径,并需要满足条件:
- 必须经过图中的绿色节点 N7、N12;
- 必须经过图中的两段绿色路段 (N2, N4)、(N13, N14);
- 必须避开图中的红色路段 (N11, N12);
- 求花费最少的最优路径。
说明:本案例来自西安邮电大学(XUPT)第12届数学建模竞赛赛题,本文进行了改编。
2.2 图的创建和可视化
2.2.1 Python 例程(NetworkX)
import numpy as np
import matplotlib.pyplot as plt
import networkx as nx
gAnt = nx.Graph()
gAnt.add_weighted_edges_from([(0,1,3),(0,2,1),(0,3,1),
(1,2,1),(1,4,1),(1,9,4),
(2,3,1),(2,4,2),(2,5,1),
(3,5,2),(3,6,2),(3,7,1),
(4,5,1),(4,9,1),
(5,6,1),(5,9,3),(5,10,1),(5,12,3),
(6,7,1),(6,8,2),(6,12,2),(6,13,4),(6,14,3),
(7,8,1),
(8,14,1),(8,15,3),
(9,10,1),(9,11,1),
(10,11,1),(10,12,2),
(11,12,1),(11,16,1),
(12,13,2),(12,16,1),
(13,14,1),(13,15,2),(13,16,2),(13,17,1),
(14,15,1),
(15,17,4),
(16,17,1)])
pos={0:(0,8),1:(7,12),2:(6,9),3:(5,6),4:(11,10),5:(14,8),
6:(17,6),7:(10,4),8:(19,4),9:(18,12),10:(21,10),11:(28,12),
12:(25,8),13:(30,7),14:(24,5),15:(29,4),16:(32,10),17:(37,8)}
fig, ax = plt.subplots(figsize=(9, 6))
nx.draw(gAnt, pos, with_labels=True, node_color='cyan', alpha=0.8)
labels = nx.get_edge_attributes(gAnt,'weight')
nx.draw_networkx_edge_labels(gAnt,pos,edge_labels=labels, font_color='m')
nx.draw_networkx_nodes(gAnt,pos,nodelist=[0,17],node_color='yellow')
nx.draw_networkx_nodes(gAnt,pos,nodelist=[7,12],node_color='lime')
nx.draw_networkx_edges(gAnt,pos,edgelist=[(2,4),(13,14)],edge_color='lime',width=3)
nx.draw_networkx_edges(gAnt,pos,edgelist=[(11,12)],edge_color='r',width=3)
plt.show()
2.2.2 程序说明
本段程序绘制网络图,包括顶点、边、边的权值,特殊顶点和特殊边的颜色设置。
- 图的创建。本例使用 nx.Graph() 创建无向图,然后用 gAnt.add_weighted_edges_from() 函数以列表向图中添加多条赋权边,每个赋权边以元组 (node1,node2,weight) 表示。
- 图的绘制。使用 nx.draw() 绘图时,默认的节点位置并不理想,可以使用 pos 属性参数指定节点位置。pos 为字典数据类型,按 node:(x_pos,y_pos) 格式设置节点位置。
- 显示边的权值。使用 nx.draw_networkx_edge_labels() 可以绘制边的属性,例程显示权值属性 ‘weight’。
- 设置顶点属性。nx.draw_networkx_nodes() 可以设置顶点的属性,例如对 nodelist 列表中的节点设置颜色属性 node_color。
- 设置边的属性。nx.draw_networkx_edges() 可以设置边的属性,例如对 edgelist 列表中的边设置线宽属性 width 和颜色属性 edge_color。
3. NetworkX 求解条件最短路径问题
3.1 无限制条件的最短路径
Python 例程(NetworkX):
minWPath1 = nx.dijkstra_path(gAnt, source=0, target=17)
lMinWPath1 = nx.dijkstra_path_length(gAnt, source=0, target=17)
print("\n问题1: 无限制条件")
print("N0 到 N17 的最短加权路径: ", minWPath1)
print("N0 到 N17 的最短加权路径长度: ", lMinWPath1)
运行结果:
问题1: 无限制条件
N0 到 N17 的最短加权路径: [0, 2, 5, 10, 11, 16, 17]
N0 到 N17 的最短加权路径长度: 6
程序说明:
- 对于无限制条件的最短路径问题,NetworkX 提供了 Dijkstra 算法、Bellman-Ford 算法、Floyd 算法和启发式算法 A* 的函数。
- 例程使用 nx.dijkstra_path() 和 nx.dijkstra_path_length() 调用 Dijkstra 算法求两个指定顶点之间的最短加权路径和最短加权路径长度。
3.2 限制条件:禁止点或禁止边
程序说明:
- 禁止点或者禁止边的处理比较简单,从图中删除对应的禁止顶点或禁止边即可。当然,也可以在创建图时就不添加这些禁止点和禁止边,但这会导致绘图时也无法反映这些顶点和边。
- 使用 remove_node(n) 删除指定顶点 n,remove_edge(u,v) 删除指定的边 (u,v)。
- 使用 remove_nodes_from([n1,…nk]) 删除多个顶点,remove_edges_from([(u1,v1),…(uk,vk)]) 删除多条边。
- 本例程中删除的点和边与 2.1 的问题描述中的要求不一致,是为了示例删除函数的使用。下同。
Python 例程:
gAntF = gAnt.copy()
gAntF.remove_node(5)
gAntF.remove_edges_from([(11,12), (13,17)])
minWPath2 = nx.dijkstra_path(gAntF, source=0, target=17)
lMinWPath2 = nx.dijkstra_path_length(gAntF, source=0, target=17)
print("\n问题2: 禁止点或禁止边的约束")
print("N0 到 N17 的最短加权路径: ", minWPath2)
print("N0 到 N17 的最短加权路径长度: ", lMinWPath2)
运行结果:
问题2: 禁止点或禁止边的约束
N0 到 N17 的最短加权路径: [0, 3, 6, 12, 16, 17]
N0 到 N17 的最短加权路径长度: 7
3.3 限制条件:一个必经点
程序说明:
- 当限制条件为一个必经点时,可以把原问题分解为两个子问题:子问题 1 为起点至必经点,子问题 2 为必经点至终点。
- 对两个子问题分别求其最短加权路径和最短加权路径长度,然后将两个子问题的结果合并,就得到经过必经点的原问题的最短加权路径和最短加权路径长度。
Python 例程:
gAntP = gAnt.copy()
minWPath3a = nx.dijkstra_path(gAntP, source=0, target=6)
lMinWPath3a = nx.dijkstra_path_length(gAntP, source=0, target=6)
minWPath3b = nx.dijkstra_path(gAntP, source=6, target=17)
lMinWPath3b = nx.dijkstra_path_length(gAntP, source=6, target=17)
minWPath3a.extend(minWPath3b[1:])
print("\n问题3: 一个必经点(N6)的约束")
print("N0 经 N6 到 N17 的最短加权路径: ", minWPath3a)
print("N0 经 N6 到 N17 的最短加权路径长度: ", lMinWPath3a+lMinWPath3b)
运行结果:
问题3: 一个必经点(N6)的约束
N0 经 N6 到 N17 的最短加权路径: [0, 3, 6, 12, 16, 17]
N0 经 N6 到 N17 的最短加权路径长度: 7
3.4 限制条件:多个必经点(方案一)
程序说明:
- 当限制条件为两个或多个必经点,如果指定这些必经点的先后顺序,可以按 3.3 的方法将原问题分解为多个子问题。
- 如果不指定这些比较点的先后顺序, 从起点出发不知道应该先去哪个必经点,这是有两种处理方法:一是用穷举法对所有可能的顺序都进行计算,然后进行比较;二是通过约束条件对所有简单路径进行判断。
- NetworkX 提供了 all_simple_paths() 函数,可以生成两个顶点之间的所有简单路径。利用简单路径算法,可以通过对约束条件的判断来求解带有多个顶点约束的最短路径问题。
- 程序实现的步骤包括:
- (1)生成指定起点、终点的所有简单路径;
- (2)判断路径是否满足包括所有必经点的限制条件;
- (3)在满足限制条件的简单路径中找加权长度最短的路径;
- (4)求最短路径的加权路径长度。
- 本段例程非常简练,综合使用了几种 Python 语言循环、判断结构的简洁写法,已经逐句注释分析,读者需要认真研读。
Python 例程:
gAntM = gAnt.copy()
minWPath4 = min([path
for path in nx.all_simple_paths(gAntM, 0, 17)
if all(n in path for n in (7, 15))],
key=lambda x: sum(gAntM.edges[edge]['weight'] for edge in nx.utils.pairwise(x)))
lenPath4 = sum(gAntM.edges[edge]['weight'] for edge in nx.utils.pairwise(minWPath4))
print("\n问题4: 多个必经点(N7, N15)的约束")
print("N0 经 N7, N15 到 N17 的最短加权路径: ", minWPath4)
print("N0 经 N7, N15 到 N17 的最短加权路径长度: ", lenPath4)
运行结果:
问题4: 多个必经点(N7, N15)的约束
N0 经 N7, N15 到 N17 的最短加权路径: [0, 3, 7, 8, 14, 15, 13, 17]
N0 经 N7, N15 到 N17 的最短加权路径长度: 8
3.5 限制条件:多个必经点(方案二)
程序说明
- 本例与 3.5 的问题实际上是相同的。限制条件都是多个必经顶点 N7、N15,解决方案都是使用 all_simple_paths() 函数生成两个顶点间的所有简单路径,程序实现的步骤也是类似的。
- 本方案按照典型的循环、判断结构的写法,便于阅读和理解。此外,如果还有其它约束条件或子任务需要在循环中处理,这样的结构更容易实现。
Python 例程
gAntM = gAnt.copy()
lMinWPath5 = minWPath5 = 1e9
for path in nx.all_simple_paths(gAntM, 0, 17):
if all(n in path for n in (7,15)):
lenPath = sum(gAntM.edges[edge]['weight'] for edge in nx.utils.pairwise(path))
if lenPath < lMinWPath5:
lMinWPath5 = lenPath
minWPath5 = path
print("\n问题5: 多个必经点(N7, N15)的约束")
print("N0 经 N7, N15 到 N17 的最短加权路径: ", minWPath5)
print("N0 经 N7, N15 到 N17 的最短加权路径长度: ", lMinWPath5)
运行结果
问题5: 多个必经点(N7, N15)的约束
N0 经 N7, N15 到 N17 的最短加权路径: [0, 3, 7, 8, 14, 15, 13, 17]
N0 经 N7, N15 到 N17 的最短加权路径长度: 8
3.6 限制条件:必经边
程序说明
- 延续处理多个必经点的思路,生成两个顶点之间的所有简单路径,通过对必经边约束条件的判断来求解必经边约束最短路径问题。
- 本例程的框架和步骤同 3.5,按照典型的循环、判断结构的写法,便于阅读和理解。
- all(n in path for n in (2,4,13,14)) 的作用,是判断路径中是否包括必经边 (2,4)、(13,14) 的各顶点,这不仅可以减小计算量,而且能确保下面使用 index() 查找顶点位置时不会发生错误。
Python 例程
gAntE = gAnt.copy()
lMinWPath6 = minWPath6 = 1e9
for path in nx.all_simple_paths(gAntE, 0, 17):
if all(n in path for n in (2,4,13,14)):
p1 = path.index(2)
if (path[p1-1]!=4 and path[p1+1]!=4): continue
p2 = path.index(13)
if (path[p2-1]!=14 and path[p2+1]!=14): continue
lenPath = sum(gAntE.edges[edge]['weight'] for edge in nx.utils.pairwise(path))
if lenPath < lMinWPath6:
lMinWPath6 = lenPath
minWPath6 = path
print("\n问题6: 必经边 (N2,N4), (N13,N14) 的约束")
print("N0 到 N17 的最短加权路径: ", minWPath6)
print("N0 到 N17 的最短加权路径长度: ", lMinWPath6)
运行结果
问题6: 必经边 (N2,N4), (N13,N14) 的约束
N0 到 N17 的最短加权路径: [0, 2, 4, 5, 6, 7, 8, 14, 13, 17]
N0 到 N17 的最短加权路径长度: 10
4. NetworkX 求解蚂蚁最优路径问题
4.1 程序说明
- 对于必经点的处理,实际上还可以有更好的方法,即结合 Dijkstra 算法的实现过程, 将限制条件作为缩小搜索空间的条件,可以降低算法的复杂度。但对于多个必经边来说,很难以此来改进基础的无约束算法,通常的处理方法就是在算法中增加一个判断是否满足约束条件的过程。
- 本例程生成两个顶点之间的所有简单路径,通过对各种约束条件的判断来求解必经边约束最短路径问题,可以同时处理禁止点、禁止边、必经点、必经边的约束条件。
- 本例程对应案例中的各项约束条件: 必须经过图中的绿色节点;必须经过图中的两段绿色路段;必须避开图中的红色路段;尽可能以最少的花费到达终点。
- 本例程是一个遍历简单路径、判断约束条件的通用框架。
- all(n in path for n in (2,4,7,12,13,14)) 的作用,一是判断路径中是否包括必经点 N7、N12;二是判断路径中是否包括必经边 (2,4)、(13,14) 的各顶点,这不仅可以减小计算量,而且能确保下面使用 index() 查找顶点位置时不会发生错误。
4.2 Python 例程
import numpy as np
import matplotlib.pyplot as plt
import networkx as nx
gAnt = nx.Graph()
gAnt.add_weighted_edges_from([(0,1,3),(0,2,1),(0,3,1),
(1,2,1),(1,4,1),(1,9,4),
(2,3,1),(2,4,2),(2,5,1),
(3,5,2),(3,6,2),(3,7,1),
(4,5,1),(4,9,1),
(5,6,1),(5,9,3),(5,10,1),(5,12,3),
(6,7,1),(6,8,2),(6,12,2),(6,13,4),(6,14,3),
(7,8,1),
(8,14,1),(8,15,3),
(9,10,1),(9,11,1),
(10,11,1),(10,12,2),
(11,12,1),(11,16,1),
(12,13,2),(12,16,1),
(13,14,1),(13,15,2),(13,16,2),(13,17,1),
(14,15,1),
(15,17,4),
(16,17,1)])
pos={0:(0,8),1:(7,12),2:(6,9),3:(5,6),4:(11,10),5:(14,8),
6:(17,6),7:(10,4),8:(19,4),9:(18,12),10:(21,10),11:(28,12),
12:(25,8),13:(30,7),14:(24,5),15:(29,4),16:(32,10),17:(37,8)}
fig, ax = plt.subplots(figsize=(9, 6))
nx.draw(gAnt, pos, with_labels=True, node_color='cyan', alpha=0.8)
labels = nx.get_edge_attributes(gAnt,'weight')
nx.draw_networkx_edge_labels(gAnt,pos,edge_labels=labels, font_color='m')
nx.draw_networkx_nodes(gAnt,pos,nodelist=[0,17],node_color='yellow')
nx.draw_networkx_nodes(gAnt,pos,nodelist=[7,12],node_color='lime')
nx.draw_networkx_edges(gAnt,pos,edgelist=[(2,4),(13,14)],edge_color='lime',width=3)
nx.draw_networkx_edges(gAnt,pos,edgelist=[(11,12)],edge_color='r',width=3)
gAntS = gAnt.copy()
gAntS.remove_edge(11,12)
lMinWPath = minWPath = 1e9
for path in nx.all_simple_paths(gAntS, 0, 17):
if all(n in path for n in (2,4,7,12,13,14)):
p1 = path.index(2)
if (path[p1-1]!=4 and path[p1+1]!=4): continue
p2 = path.index(13)
if (path[p2-1]!=14 and path[p2+1]!=14): continue
lenPath = sum(gAntS.edges[edge]['weight'] for edge in nx.utils.pairwise(path))
if lenPath < lMinWPath:
lMinWPath = lenPath
minWPath = path
print("\n蚂蚁最优路径问题(带有禁止点、禁止边、必经点、必经边的约束条件)")
print("约束条件:必经点 N7,N12,必经边 (N2,N4), (N13,N14),禁止边 (11,12)")
print("N0 到 N17 的最短加权路径: ", minWPath)
print("N0 到 N17 的最短加权路径长度: ", lMinWPath)
edgeList = []
for i in range(len(minWPath)-1):
edgeList.append((minWPath[i],minWPath[i+1]))
nx.draw_networkx_edges(gAnt,pos,edgelist=edgeList,edge_color='b',width=4)
plt.show()
4.3 运行结果
蚂蚁最优路径问题(带有禁止点、禁止边、必经点、必经边的约束条件)
约束条件:必经点 N7,N12,必经边 (N2,N4), (N13,N14),禁止边 (11,12)
N0 到 N17 的最短加权路径: [0, 2, 4, 5, 6, 7, 8, 14, 13, 12, 16, 17]
N0 到 N17 的最短加权路径长度: 13
5. 总结
- 数模竞赛中的最短路径问题,往往是带有约束条件的最短路径问题。
- 很多计算机语言算法工具包(包括但不限于 Python)都对算法进行了封装,没有提供设置约束条件的选项和接口,因此用户不能把条件判断语句加入这些封装函数的程序内部。
- 本文基于 NetworkX 工具包,建立了一个遍历简单路径、判断约束条件的通用框架,不仅可以处理禁止点、禁止边、必经点、必经边的约束条件处理,而且可以扩展到任何约束条件。
【本节完】
版权声明:
欢迎关注『Python小白的数学建模课 @ Youcans』 原创作品
原创作品,转载必须标注原文链接:(https://blog.csdn.net/youcans/article/details/118566373)。
Copyright 2021 Youcans, XUPT
Crated:2021-07-10
欢迎关注 『Python小白的数学建模课 @ Youcans』 系列,持续更新 Python小白的数学建模课-01.新手必读 Python小白的数学建模课-02.数据导入 Python小白的数学建模课-03.线性规划 Python小白的数学建模课-04.整数规划 Python小白的数学建模课-05.0-1规划 Python小白的数学建模课-06.固定费用问题 Python小白的数学建模课-07.选址问题 Python小白的数学建模课-09.微分方程模型 Python小白的数学建模课-10.微分方程边值问题 Python小白的数学建模课-12.非线性规划 Python小白的数学建模课-15.图论的基本概念 Python小白的数学建模课-16.最短路径算法 Python小白的数学建模课-17.条件最短路径算法 Python小白的数学建模课-A1.国赛赛题类型分析 Python小白的数学建模课-A2.2021年数维杯C题探讨 Python小白的数学建模课-A3.12个新冠疫情数模竞赛赛题及短评 Python小白的数学建模课-B2. 新冠疫情 SI模型 Python小白的数学建模课-B3. 新冠疫情 SIS模型 Python小白的数学建模课-B4. 新冠疫情 SIR模型 Python小白的数学建模课-B5. 新冠疫情 SEIR模型 Python小白的数学建模课-B6. 新冠疫情 SEIR改进模型 Python数模笔记-PuLP库 Python数模笔记-StatsModels统计回归 Python数模笔记-Sklearn Python数模笔记-NetworkX Python数模笔记-模拟退火算法
|