bellman-ford算法可以求解图中任意两点之间只经过k条边的最短路径长度,且权值可以为负数。
给定一个 n 个点,m条边的图,求最多经过 k 条边,1 号点到 n 号点的最短路径距离,bellman-ford的步骤是:
初始情况为所有点距离源点的距离为 INF,源点距离源点的距离为 0;
枚举 k 次,在每次枚举中,枚举所有边,由上次枚举得到的结果根据边权更新所有点到源点的最短距离;
时间复杂度为O (nm) ;
bellman-ford不受重边和负环的影响;
对于重边,能遍历到最短的那条即可;
对于负环,k 次循环代表最多经过 k 条边,由于是有步数限制的,负环不会造成死循环。
图示:
代码模板如下:
#include<iostream>
#include<cstring>
using namespace std;
const int M=510,N=1e4+10;
const int INF=0x3f3f3f3f;
struct Edge{//边
int a,b,w;
}edges[N];
int d[M],backup[M];//距离数组与其备份
int n,m,k;//n点m边k步
int bellman_ford(){
//初始化距离
memset(d,0x3f,sizeof d);
d[1]=0;
for(int i=0;i<k;i++){//枚举n次
memcpy(backup,d,sizeof d);//将d拷贝到backup中
for(int j=0;j<m;j++){//每次枚举m条边
int l=edges[j].a,r=edges[j].b,c=edges[j].w;//取出边
d[r]=min(d[r],backup[l]+c);//在backup上更新d
}
}
if(d[n]>INF/2)return INF;
return d[n];
}
int main(){
cin>>n>>m>>k;
int l,r,c,i=0;
for(int i=0;i<m;i++){
cin>>l>>r>>c;
edges[i]={l,r,c};
}
int t=bellman_ford();
if(t==INF)cout<<"impossible";
else cout<<t;
return 0;
}
由于更新操作是枚举所有边,只需用结构体将所有边存储即可;
由于存在负权边,权值为负的边会把距离为INF的点更新掉,最后判断源点到 n 点是否有最短路的条件不是 n 点距离源点的距离否等于 INF,而是 n 点距离源点的距离是否大于 INF/2 ;
每次迭代中,先将图复制一份,用复制的图来更新原图中点的最短距离,防止串联情况的发生。