刚刚学了Bellmanford算法,于是找了道做过的模板题来试试手。当时直接用的模板,今天靠理解手打了一遍Bellmanford,进一步加深了理解。
Bellmanford算法,简而言之就是对一张有n个点的图的所有边,进行n-1次松弛操作。然后判断,如果仍然能进行第n次松弛操作,说明图里有负环,可以沿着这个负环不停地走下去让最短路径变为无穷小。
建图用的链式前向星。
前M条权值为正的边是双向边,后W条权值为负的是单向的。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
const int MAXN=1000;
int dis[MAXN];
int head[MAXN];
struct edge{
int to,len,next;
}E[8000];
bool bellmanford(int n){
for(int i=2;i<=n;i++) dis[i]=INF;
dis[1]=0;
for(int i=1;i<=n-1;i++) //对链式前向星中共2M+W条边各执行n-1次松弛操作
{
for(int j=1;j<=n;j++){
if(dis[j]==INF) continue;
for(int k=head[j];k!=0;k=E[k].next){ //链式前向星编号从1开始,故以K!=0结束,若从0开始,改为k!=-1结束
if(E[k].len!=INF&&dis[E[k].to]>(dis[j]+E[k].len)){
dis[E[k].to]=dis[j]+E[k].len;
}
}
}
}
for(int j=1;j<=n;j++){ //若经过以上n-1次松弛以后仍然能松弛,说明图中存在权值为负的环,无最短路径
for(int k=head[j];k!=0;k=E[k].next){
if(E[k].len!=INF&&dis[E[k].to]>(dis[j]+E[k].len)){
return false; //存在负环
}
}
}
return true; //无负环
}
int main()
{
int F,N,M,W,i,j,k;
scanf("%d",&F);
while(F--)
{ memset(head,0,sizeof(head));
scanf("%d%d%d",&N,&M,&W);
memset(E,0,sizeof(E));
memset(dis,0,sizeof(dis));
for(i=1;i<=M;i++) //建立了编号从1到2M,共2M条边(前M条与后M条一一对应为反向边)
{ int index;
scanf("%d%d%d",&index,&E[i].to,&E[i].len);
E[i].next=head[index];
head[index]=i;
E[i+M].to=index;
E[i+M].len=E[i].len;
E[i+M].next=head[E[i].to];
head[E[i].to]=i+M;
}
for(i=2*M+1;i<=2*M+W;i++)
{ int index;
scanf("%d%d%d",&index,&E[i].to,&E[i].len); //建立了编号从2M+1到2M+W共W条边
E[i].len=E[i].len*(-1);
E[i].next=head[index];
head[index]=i;
}
/* for(int j=1;j<=N;j++){ //测试,检查建边情况
for(int k=head[j];k!=0;k=E[k].next){
cout<<j<<" "<<E[k].to<<" "<<E[k].len<<endl;
}
} */
if(bellmanford(N))
{
printf("NO\n");
}
else
printf("YES\n");
}
return 0;}