【POJ】3259 Wormholes bellman-ford | SPFA

诶，真的是越活越回去了，现在打的程序还不如以前了，慢的爆炸……（先测的是bellman-ford，后测的是SPFA）



题目大意：FJ想要进行时光旅行，现在有F个农场，对于每个农场，已知该农场有N片农田，M条小路，W个虫洞。其中小路是双向的，虫洞是单向的。对于每条小路，有S,E,T三个参数：S和E表示小咯两端农田的序号，通过这条小咯需要的T的时间。对于每个虫洞，同样有S,E,T三个参数：S表示虫洞起点所在的农田的序号，E表示虫洞终点所在的农田的序号，通过这个虫洞需要-T的时间。现在FJ想知道他在每个农场是否能回到从前。

我们从题目中可以清晰的看出，题目就是想让我们求每个农场是否存在负权环。一道大水题。

方法一：bellman-ford，用side数组来存边。若松弛次数超过n次，就说明存在负权环。

#include <cstdio>
#include <vector>
#include <cstring>
using namespace std;

struct note{
int x,y,w;
}side[2710];
int ti,n,m,p,x,y,w,dis[510],g;
bool b;

int main(void){
scanf("%d",&ti);
while (ti--){
scanf("%d%d%d",&n,&m,&p);
for (int i=1; i<=m; ++i)
scanf("%d%d%d",&side[i].x,&side[i].y,&side[i].w);
for (int i=m+1; i<=m+p; ++i)
scanf("%d%d%d",&side[i].x,&side[i].y,&side[i].w),side[i].w*=-1;
memset(dis,0x3f,sizeof(dis));
b=1,g=0;dis[1]=0;
while (b){
b=0;
if (++g>n) break;
for (int i=1; i<=m; ++i){
if (dis[side[i].y]>dis[side[i].x]+side[i].w)
dis[side[i].y]=dis[side[i].x]+side[i].w,b=1;
if (dis[side[i].x]>dis[side[i].y]+side[i].w)
dis[side[i].x]=dis[side[i].y]+side[i].w,b=1;
}
for (int i=m+1; i<=m+p; ++i)
if (dis[side[i].y]>dis[side[i].x]+side[i].w)
dis[side[i].y]=dis[side[i].x]+side[i].w,b=1;
}
if (g>n) printf("YES\n");
else printf("NO\n");
}
return 0;
}
方法二：SPFA——bellman-Ford的队列优化升级版，可惜在这题中慢的起飞……（好像是因为SPFA计算了许多不必要的路线）若一个节点入队超过n次就说明存在负权环。
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;

queue <int> que;
int t,n,m1,m2,x,y,w,map[510][510],c[510],dis[510];
bool b,bo[510];

bool spfa(int s){
while (!que.empty()) que.pop();
memset(c,0,sizeof c);
memset(bo,0,sizeof bo);
memset(dis,0x3f,sizeof dis);
que.push(s);c[s]=1;bo[s]=1;
while (!que.empty()){
int p=que.front();que.pop();
bo[p]=0;
for (int i=1; i<=n; ++i)
if (dis[i]>dis[p]+map[p][i]){
dis[i]=dis[p]+map[p][i];
if (!bo[i]){
if (++c[i]>n) return 1;
bo[i]=1;
que.push(i);
}
}
}
return 0;
}

int main(void){
scanf("%d",&t);
while (t--){
scanf("%d%d%d",&n,&m1,&m2);
memset(map,0x3f,sizeof map);
for (int i=1; i<=m1; ++i){
scanf("%d%d%d",&x,&y,&w);
if (w<map[x][y]) map[x][y]=map[y][x]=w;
}
for (int i=1; i<=m2; ++i){
scanf("%d%d%d",&x,&y,&w);
map[x][y]=-w;
}
b=0;
for (int i=1; i<=n; ++i)
if (spfa(i)){
b=1;
break;
}
if (b) printf("YES\n");
else printf("NO\n");
}
return 0;
}
写在最后：虽然SPFA的运行速度过慢可能跟计算了许多不必要的路线有关，但是和我用的是邻接矩阵有不可分割的关系。在以后的做题过程中，尽量用高级一些的数据结构，狮子搏兔亦用全力。