【POJ】3259 Wormholes

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

题意：n个农场，m条双向路径，w条单向路径（虫洞）。单向虫洞路径是负值。农夫想知道自己能不能看到自己（X）。

题解：其实刚开始没太读懂题意。然后其实如果他能看到自己，说明已经通过虫洞形成了一个负环。也就是通过spfa寻找负环（负权回路）。这里的判断就是加一个cnt[]数组记录该点的入队次数，大于等于n说明已经形成负环。

代码：

 #include<iostream>
 #include<cstdio>
 #include<stack>
 #include<vector>
 #include<queue>
 #include<algorithm>
 using namespace std;
 const int maxn = 2e5+;
 const int inf = 0xffffff;
 vector< pair<int,int> > e[maxn];
 
 int n,m,w;
 int d[maxn],inq[maxn];
 int cnt[maxn];
 
 void init(){
     for(int i = ; i <= n; i++)
         e[i].clear();
 }
 
 void spfa(){
     for(int i =  ;i <= n ; i++)
         inq[i] = ,d[i] = inf,cnt[i] = ;
     queue<int>Q;
     Q.push();
     d[] = ;
     inq[] = ;
     cnt[] = ;
     while( !Q.empty() ){
         int u = Q.front();
         Q.pop();
         inq[u] = ;
         for(int i = ; i < e[u].size() ; i++){
             int v = e[u][i].first;
             int val = e[u][i].second;
             if(d[v] > d[u] + val){
                 d[v] = d[u] + val;
                 if(inq[v] == ){
                     cnt[v]++;
                     inq[v] = ;
                     if(cnt[v] >= n){    //判定一点入队次数大于总顶点数，存在负环
                         printf("YES\n");
                         return;
                     }
                     Q.push(v);
                 }
             }
         }
 
     }
     printf("NO\n");
 }
 int main() {
     int F;
     cin>>F;
     while(F--){
         cin>>n>>m>>w;
         init();
         //双向正权
         int x,y,z;
         for(int i = ; i <= m ;i++){
             cin>>x>>y>>z;
             e[x].push_back(make_pair(y,z));
             e[y].push_back(make_pair(x,z));
         }
         //单向负权
         for(int i = ; i <= w; i++){
             cin>>x>>y>>z;
             e[x].push_back(make_pair(y,-z));
         }
         spfa();
     }
 
     return ;
 }