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 + ;
 const int INF = 1e9;
 struct data {
     int next , to , cost;
 }edge[MAXN * ];
 int head[MAXN] , d[MAXN] , cnt[MAXN] , cont;
 bool vis[MAXN];

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

 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] = ;
     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( , n)) {
             printf("YES\n");
         }
         else {
             printf("NO\n");
         }
     }
 }