poj3259 Wormholes（Bellman-Ford判断负圈）

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
#define IO ios::sync_with_stdio(false);cin.tie(0);
#define INF 0x3f3f3f3f
typedef unsigned long long ll;
using namespace std;
int a[1010][1010], dist[1010];
int k, kase, n, m, w, s, e, t;
typedef struct{
    int from, to;
    int cost;
}Node;
Node node[10010];
int solve()
{ 
    memset(dist, 0, sizeof(dist));//可以检查出所有的负圈 
    for(int i = 1; i <= n; i++){
        for(int j = 0; j < 2*m+w; j++){
            Node e = node[j];
            if(dist[e.to] > dist[e.from]+e.cost){
                dist[e.to] = dist[e.from]+e.cost;
                if(i == n) return 1;
            }
        }
    }
    return 0;
}
int main()
{
    scanf("%d", &kase);
    while(kase--){ 
        scanf("%d%d%d", &n, &m, &w); 
        for(int i = 0; i < 2*m; i++){
            scanf("%d%d%d", &s, &e, &t);//路是双向的 
            node[i].from = s; node[i].to = e;
            node[i].cost = t;
            i++;
            node[i].from = e; node[i].to = s;
            node[i].cost = t;
        }
        for(int i = 2*m; i < 2*m+w; i++){
            scanf("%d%d%d", &s, &e, &t);//虫洞是单向的    
            node[i].from = s; node[i].to = e;
            node[i].cost = -t;            
        }
        if(solve()) cout << "YES" << endl;
        else cout << "NO" << endl; 
    } 
    return 0;
}