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;
}