【USACO06DEC】虫洞 G ,C++ 代码实现

题目描述：

John 在他的农场中闲逛时发现了许多虫洞。虫洞可以看作一条十分奇特的有向边，并可以使你返回到过去的一个时刻（相对你进入虫洞之前）。 John 的每个农场有 m 条小路（无向边）连接着 n 块地（从 1∼n 标号），并有 w 个虫洞。

现在 John 希望能够从某块地出发，走过一条路径回到出发点，且同时也回到了出发时刻以前的某一时刻。请你告诉他能否做到。

输入：

第一行：N,K。 接下来一行 K 个整数，第 i 个为 Xi。 第 N+2 行：Q。 接下来 Q 行：每行两个整数 Li, Ri。

输出：

对于每组测试数据，输出一行一个字符串，如果能回到出发时刻之前，则输出 YES，否则输出 NO。

样例输入：

2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8

样例输出：

NO
YES

提示：

样例 2 解释

John 只需要沿着 1→2→3→1 的路径一直转圈即可，每转完一圈，时间就会减少一秒。

数据范围与约定

对于 100% 的数据，1≤T≤5，1≤n≤500，1≤m≤2500，1≤w≤200，1≤p≤104。

#include<bits/stdc++.h>
using namespace std;
struct Node{
	int B,E,dis;
}edge[5202];//存储边的结构体 
int T,n,m,w,cnt,d[5202];
inline bool Judge(){
	for(int i=2;i<=n;i++)
		d[i]=123456789;
	d[1]=0;
	//SPFA最短路算法判断负环 
	for(int i=1;i<=n;i++)
		for(int j=1;j<=cnt;j++)
			if(d[edge[j].E]>d[edge[j].B]+edge[j].dis){//成功回到过去 
				d[edge[j].E]=d[edge[j].B]+edge[j].dis;
				if(i==n)//成功走完整个农场并回到过去 
					return 1;
			}
	return 0;
}
int main(){
	scanf("%d",&T);//T个农场 
	while(T--){
		scanf("%d%d%d",&n,&m,&w);
		cnt=0;
		for(int i=1,s,e,t;i<=m;i++){
			scanf("%d%d%d",&s,&e,&t);
			edge[++cnt].B=s,edge[cnt].E=e,edge[cnt].dis=t;//连一条从s到e,边权为t的边 
			edge[++cnt].B=e,edge[cnt].E=s,edge[cnt].dis=t;//路是双向边 
		}
		for(int i=1,s,e,t;i<=w;i++){
			scanf("%d%d%d",&s,&e,&t);
			edge[++cnt].B=s,edge[cnt].E=e,edge[cnt].dis=-t;//虫洞可以回到过去,而不能去到未来,所以这是一条边权为-t的单向边 
		}
		if(Judge())//成功
			printf("YES\n"); 
		else
			printf("NO\n");
	}
}