POJ 3259 Wormholes(SPFA判负环)

题目链接：http://poj.org/problem?id=3259

题目大意是给你n个点，m条双向边，w条负权单向边。问你是否有负环(虫洞)。

这个就是spfa判负环的模版题，中间的cnt数组就是记录这个点松弛进队的次数，次数超过点的个数的话，就说明存在负环使其不断松弛。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 1e3 + 5;
const int INF = 1e9;
struct data {
    int next , to , cost;
}edge[MAXN * 10];
int head[MAXN] , d[MAXN] , cnt[MAXN] , cont;
bool vis[MAXN];

void init(int n) {
    for(int i = 1 ; i <= n ; i++) {
        d[i] = INF;
        cnt[i] = 0;
        head[i] = -1;
        vis[i] = false;
    }
    cont = 0;
}

inline void add(int u , int v , int cost) {
    edge[cont].next = head[u];
    edge[cont].to = v;
    edge[cont].cost = cost;
    head[u] = cont++;
}

bool spfa(int s , int n) {
    d[s] = 0;
    queue <int> que;
    while(!que.empty()) {
        que.pop();
    }
    que.push(s);
    while(!que.empty()) {
        int temp = que.front();
        que.pop();
        vis[temp] = false;
        for(int i = head[temp] ; ~i ; i = edge[i].next) {
            int v = edge[i].to;
            if(d[v] > d[temp] + edge[i].cost) {
                d[v] = d[temp] + edge[i].cost;
                if(!vis[v]) {
                    que.push(v);
                    vis[v] = true;
                    cnt[v]++;
                    if(cnt[v] >= n)
                        return true;
                }
            }
        }
    }
    return false;
}

int main()
{
    int n , m , c , u , v , cost , t;
    scanf("%d" , &t);
    while(t--) {
        scanf("%d %d %d" , &n , &m , &c);
        init(n);
        while(m--) {
            scanf("%d %d %d" , &u , &v , &cost);
            add(u , v , cost);
            add(v , u , cost);
        }
        while(c--) {
            scanf("%d %d %d" , &u , &v , &cost);
            add(u , v , -cost);
        }
        if(spfa(1 , n)) {
            printf("YES
");
        }
        else {
            printf("NO
");
        }
    }
}