【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的运行速度过慢可能跟计算了许多不必要的路线有关，但是和我用的是邻接矩阵有不可分割的关系。在以后的做题过程中，尽量用高级一些的数据结构，狮子搏兔亦用全力。