【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+7;
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 = 0; i <= n; i++)
        e[i].clear();
}

void spfa(){
    for(int i = 0 ;i <= n ; i++)
        inq[i] = 0,d[i] = inf,cnt[i] = 0;
    queue<int>Q;
    Q.push(1);
    d[1] = 0;
    inq[1] = 1;
    cnt[1] = 1;
    while( !Q.empty() ){
        int u = Q.front();
        Q.pop();
        inq[u] = 0;
        for(int i = 0; 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] == 0){
                    cnt[v]++;
                    inq[v] = 1;
                    if(cnt[v] >= n){    //判定一点入队次数大于总顶点数，存在负环
                        printf("YES
");
                        return;
                    }
                    Q.push(v);
                }
            }
        }

    }
    printf("NO
");
}
int main() {
    int F;
    cin>>F;
    while(F--){
        cin>>n>>m>>w;
        init();
        //双向正权
        int x,y,z;
        for(int i = 1; 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 = 1; i <= w; i++){
            cin>>x>>y>>z;
            e[x].push_back(make_pair(y,-z));
        }
        spfa();
    }

    return 0;
}