POJ 3259 - Wormholes

题目大意：有一个农夫有F个农场，每个农场里有N个农田，农田之间有M条道路，每条道路都是双向的，每条道路信息包括起点，终点，从起点到终点需要的时间。有W个虫洞，进入虫洞所在的农田就会穿越到指定的农田，并且时间回到x秒以前。农夫是个狂热的时空旅行迷，他想知道他能否借助虫洞碰到他自己。

解题思路：floyd是可以ac的，但是稍不限制就会超时。用bellman ford的算法，任意一个起点出发，虫洞就当作负权值的道路，问题就变成了是否存在负环了，注意M条道路双向的，而虫洞是单向的，数组一次输入包括了道路和虫洞，所以要开大一些。

ac代码：

#include <iostream>
#define INF 5000005
using namespace std;
int F, N, M, W, u[3005], v[3005];
int w[3005], dis[505], t1, t2, flag;
int main()
{
	scanf("%d", &F);
	while (F--){
		scanf("%d%d%d", &N, &M, &W);
		flag = 0;
		for (int i=1; i<=N; i++)
			dis[i] = INF;
		for (int i=0; i<M+W; i++){
			scanf("%d%d%d", &u[i], &v[i], &w[i]);
			if (i >= M)
				w[i] = -w[i];
		}
		dis[1] = 0;
		for (int i=0; i<N-1; i++){
			for (int j=0; j<M+W; j++){
				t1 = u[j], t2 = v[j];
				if (j < M && dis[t1] > dis[t2] + w[j])
					dis[t1] = dis[t2] + w[j];
				if (dis[t2] > dis[t1] + w[j])
					dis[t2] = dis[t1] + w[j];
			}
		}
		for (int i=0; i<M+W; i++){
			t1 = u[i], t2 = v[i];
			if (i < M && dis[t1] > dis[t2] + w[i])
				flag = 1;
			if (dis[t2] > dis[t1] + w[i])
				flag = 1;
		}
		if (flag)
			printf("YES\n");
		else
			printf("NO\n");
	}
return 0;
}