Poj_3259 Wormholes(最短路)

题意：

Farmer John拥有很多牧场，在牧场之间有双向的道路，也有单向的虫洞。虫洞可以使John从A到B并且时间后退T，John在想能否从某一点出发并在回到该点的时候时间比出发时还要走。

思路：

转化为最短路的找负环，将虫洞转化为带负权的边。考虑到双向边并且有重边，最开始想要用Floyd，但是WA掉了。后来改为Bellman-Ford，并不需要对重边进行处理。

代码实现：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAX_V = 520;
const int MAX_E = 2500*2+200;
const int INF = 1020;

struct Edge{
int from;
int to;
int cost;
};

int F;
int cnt;
bool flag;
int N,W,M;
int dis[MAX_V];
Edge edge[MAX_E];
bool find_loop();
int main()
{
scanf("%d",&F);
while( F-- ){
scanf("%d%d%d",&N,&W,&M);
int a,b,c;
cnt = 0;
flag = false;
for( int i = 0; i < W; i++ ){
scanf("%d%d%d",&a,&b,&c);
edge[cnt].from = a;
edge[cnt].to = b;
edge[cnt].cost = c;
cnt++;
edge[cnt].from = b;
edge[cnt].to = a;
edge[cnt].cost = c;
cnt++;
}
for( int i = 0; i < M; i++ ){
scanf("%d%d%d",&a,&b,&c);
edge[cnt].from = a;
edge[cnt].to = b;
edge[cnt].cost = -c;
cnt++;
}
for( int i = 1; i <= N; i++ ){
dis[i] = INF;
}
dis[1] = 0;
flag = find_loop();
if( flag == true ){
printf("YES\n");
}
else{
printf("NO\n");
}
}
return 0;
}

bool find_loop(){
for( int i = 0; i < N; i++ ){
for( int j = 0; j < cnt; j++ ){
Edge e = edge[j];
if( dis[e.to] > dis[e.from]+e.cost ){
dis[e.to] = dis[e.from]+e.cost;
if( i == N-1 ){
return true;
}
}
}
}
return false;
}