【算法】单元最短路径之Bellman-Ford算法和SPFA算法
SPFA是经过对列优化的bellman-Ford算法,因此,在学习SPFA算法之前,先学习下bellman-Ford算法。
bellman-Ford算法是一种通过松弛操作计算最短路的算法。
适用条件
1.单源最短路径(从源点s到其它所有顶点v);2.有向图&无向图(无向图可以看作(u,v),(v,u)同属于边集E的有向图);3.边权可正可负(如有负权回路输出错误提示);4.差分约束系统;
bellman-Ford的具体操作是这样的:
- 初始化,dis数组表示从起点到达第i个点的最短距离。初始化时:dis[i]=edge[起点][i];如果没有边相接,则设为MAXN;
- 循环n-1次,遍历每个边,将dis[边[i].目标节点]=min(dis[边[i].目标节点],dis[边[i].初始节点]+边[i].权值);
- 【可选】检验负权回路:判断边集合中的每一条边的两个端点是否收敛。如果存在未收敛的顶点,则算法返回false,表明问题无解;否则算法返回true,并且从源点可达的顶点v的最短距离保存在 dis[v]中。
#include <bits/stdc++.h> using namespace std; int n,m; //Bellman-Ford algorithm struct Edge { int a,b,v; }edge[500005]; long long dis[500005]; void bellman_ford(int x) { dis[x]=0; for(int i=1;i<=n-1;i++) { for(int j=1;j<=m;j++) { dis[edge[j].b]=min(dis[edge[j].b],dis[edge[j].a]+edge[j].v); } } return ; /* bool flag = 1; //判断是否含有负权回路 for(int i = 1; i <= m; ++i) if(dis[edge[i].b] > dis[edge[i].a] + edge[i].cost) { flag = 0; break; } return flag; */ } int main() { int s; cin>>n>>m>>s; for(int i=1;i<=m;i++) { scanf("%d%d%d",&edge[i].a,&edge[i].b,&edge[i].v); } for(int i=1;i<=n;i++)dis[i]=2147483647; bellman_ford(s); for(int i=1;i<=n;i++) cout<<dis[i]<<" "; return 0; }
这个算法在luogu上70分,TLE了两个测试点,因此需要学习一个更牛逼的算法,SPFA,听说这是一个交大的教授发明的。附上原论文连接:https://wenku.baidu.com/view/df249954d4d8d15abe234eff.html
SPFA算法的全称是:Shortest Path Faster Algorithm,是西南交通大学段凡丁于 1994 年发表的论文中的名字。不过,段凡丁的证明是错误的,且在 Bellman-Ford 算法提出后不久(1957 年 )已有队列优化内容,所以国际上不承认 SPFA 算法是段凡丁提出的。
为了避免最坏情况的出现,在正权图上应使用效率更高的Dijkstra算法。若给定的图存在负权边,类似Dijkstra算法等算法便没有了用武之地,SPFA算法便派上用场了。
#include <bits/stdc++.h> using namespace std; int n,m,start; long long dis[1000005],used[1000005],head[1000005]; int num_edge=0; struct Graph { int to,w,next; }graph[10000005]; bool spfa(int x) { queue <int> q; dis[x]=0; used[x]=1; q.push(x); while(!q.empty()) { int u=q.front(); q.pop(); used[u]=0; for(int i=head[u];i;i=graph[i].next) { int v=graph[i].to; long long Songchi=dis[u]+graph[i].w; if(dis[v]>Songchi) { dis[v]=Songchi; if(used[v]==0) { used[v]=1; q.push(v); } } } } return 1; } void addedge(int from,int to,int dis)//邻接表存边 { graph[++num_edge].next=head[from]; graph[num_edge].to=to; graph[num_edge].w=dis; head[from]=num_edge; } int main() { memset(head,0,sizeof(head)); memset(used,0,sizeof(used)); cin>>n>>m>>start; for(int i=1;i<=n;i++)dis[i]=2147483647; for(int i=1;i<=m;i++) { int a,b,w; scanf("%d%d%d",&a,&b,&w); addedge(a,b,w); } if(spfa(start)) { for(int i=1;i<=n;i++) cout<<dis[i]<<" "; } }