【POJ 3259】Wormholes (SPFA 判斷負環)

傳送門

    Wormholes
    題意：給出一張圖，判斷是否存在負環。

I think

    方法：SPFA求單源最短路，若一個點入隊超過Ｎ次，那麼存在負環。

Code

#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;

const int sm = 5250;
const int sn = 505;
const int Inf = 6e6;

int T,N,M,W,tot;
int hd[sn],to[sm],nxt[sm],w[sm];
int cnt[sn],dis[sn];
bool vis[sn];
queue<int>Q;
void Add(int u,int v,int c) {
    to[++tot]=v,nxt[tot]=hd[u],hd[u]=tot,w[tot]=c;
}
bool SPFA() {
    while(!Q.empty()) Q.pop();
    for(int i=1;i<=N;++i)
        dis[i]=Inf,cnt[i]=0,vis[i]=0;
    int t; dis[1]=0;Q.push(1);
    vis[1]=1;++cnt[1];
    while(!Q.empty()) {
        t=Q.front(),Q.pop();
        vis[t]=false;
        for(int i=hd[t];i;i=nxt[i])
            if(dis[to[i]]>dis[t]+w[i]) {
                dis[to[i]]=dis[t]+w[i];
                if(!vis[to[i]]) {
                    Q.push(to[i]),++cnt[to[i]];
                    vis[to[i]]=true;
                    if(cnt[to[i]]>N) return true;
                }
            }
    }
    return false;
}
void init() {
    tot=0;
    memset(hd,0,sizeof(hd));
}
int main() {
    int u,v,w;
    scanf("%d",&T);
    while(T--) {
        init();
        scanf("%d%d%d",&N,&M,&W);
        for(int i=1;i<=M;++i) {
            scanf("%d%d%d",&u,&v,&w);
            Add(u,v,w),Add(v,u,w);
        }
        for(int i=1;i<=W;++i) {
            scanf("%d%d%d",&u,&v,&w);
            Add(u,v,-w);
        }
        puts(SPFA()?"YES":"NO");
    }
    return 0;
}