负环判断poj

一、导入（“虫洞”~wormhole）

题目一 3259 -- Wormholes

//由于不知道起点，就将初始距离都设置为0，将所有点都入队。等同于加入一个超级源点 从每个点出发若能找到一条负环
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=510,M=6010;
int _,n,m,k,tot,to[M],w[M],nxt[M],h[N],d[N],cnt[N],inQ[N];
void add_edge(int a,int b,int c){
    to[++tot]=b;w[tot]=c;nxt[tot]=h[a];h[a]=tot;
}
int SPFA(){
    memset(d,0,sizeof(d));
    memset(cnt,0,sizeof(cnt));
    memset(inQ,0,sizeof(inQ));
    queue<int>Q;
    for(int i=1;i<=n;i++){Q.push(i);inQ[i]=1;}
    while(Q.size()){
        int u=Q.front();Q.pop();inQ[u]=0;
        for(int i=h[u];~i;i=nxt[i]){
            int v=to[i];
            if(d[u]+w[i]<d[v]){
                d[v]=d[u]+w[i];
                if(!inQ[v]){
                    if(++cnt[v]>=n-1)return 1;
                    Q.push(v);inQ[v]=1;
                }
            }
        }
    }
    return 0;
}
int main(){
    scanf("%d",&_);
    while(_--){
        memset(h,-1,sizeof(h));tot=0;
        scanf("%d%d%d",&n,&m,&k);
        int u,v,w;
        while(m--){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);add_edge(v,u,w);
        }
        while(k--){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,-w);
        }
        puts(SPFA()?"YES":"NO");
    }
    return 0;
}

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=510,M=6010;
int _,n,m,k,tot,to[M],w[M],nxt[M],h[N],d[N],cnt[N],inQ[N];
void add_edge(int a,int b,int c){
    to[++tot]=b;w[tot]=c;nxt[tot]=h[a];h[a]=tot;
}
int SPFA(int s){
    memset(d,0x3f,sizeof(d));
    memset(cnt,0,sizeof(cnt));
    memset(inQ,0,sizeof(inQ));
    queue<int>Q;
    Q.push(s);inQ[s]=1;cnt[s]=1;d[s]=0;
    while(Q.size()){
        int u=Q.front();Q.pop();inQ[u]=0;
        for(int i=h[u];~i;i=nxt[i]){
            int v=to[i];
            if(d[u]+w[i]<d[v]){
                d[v]=d[u]+w[i];
                if(!inQ[v]){
                    if(++cnt[v]>=n)return 1;
                    Q.push(v);inQ[v]=1;
                }
            }
        }
    }
    return 0;
}
int main(){
    scanf("%d",&_);
    while(_--){
        memset(h,-1,sizeof(h));tot=0;
        scanf("%d%d%d",&n,&m,&k);
        int u,v,w;
        while(m--){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);add_edge(v,u,w);
        }
        while(k--){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,-w);
        }
        puts(SPFA(1)?"YES":"NO");
    }
    return 0;
}

[方法1]经典方法 169ms

#include<cstdio>
#include<cstring>
#include<utility>
#include<vector>
#include<queue>
#define N 505
using namespace std;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
vector<pair<int,int> > G[N];
int T,n,m,k,d[N],cnt[N]; bool inQ[N];

int SPFA(int s){
    memset(d,0x3f,sizeof(d));
    memset(cnt,0,sizeof(cnt));
    memset(inQ,false,sizeof(inQ));
    queue<int> Q;
    Q.push(s); inQ[s]=1; cnt[s]=1; d[s]=0;
    while(!Q.empty()){
        int u=Q.front(); Q.pop(); inQ[u]=0;
        for(int i=0;i<G[u].size();i++){
            int v=G[u][i].first,w=G[u][i].second;
            if(d[u]+w<d[v]){
                d[v]=d[u]+w;
                if(!inQ[v]){
                    Q.push(v); inQ[v]=1; cnt[v]++;
                    if(cnt[v]>n-1)return 1;
                }
            }
        }
    }
    return 0;
}

int main(){
    T=read();
    while(T--){
        for(int i=0;i<N;i++)G[i].clear();
        n=read(),m=read(),k=read();
        int u,v,w;
        while(m--)
            u=read(),v=read(),w=read(),
            G[u].push_back(make_pair(v,w)),G[v].push_back(make_pair(u,w));
        while(k--)
            u=read(),v=read(),w=read(),
            G[u].push_back(make_pair(v,-w));
        puts(SPFA(1)?"YES":"NO");
    }
    return 0;
}

[方法2]32ms

#include<bits/stdc++.h>
using namespace std;
int dis[600],V[600],flag;
vector<int> G[600],W[600];
bool DFS(int u){
    V[u]=1;
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(dis[u]+W[u][i]<dis[v]){
            dis[v]=dis[u]+W[u][i];
            if(V[v]==1||DFS(v))return true;
        }
    }
    V[u]=0;
    return false;
}

int main(){
    int _;scanf("%d",&_);
    while(_--){
        int n,m,m2,u,v,w;
        scanf("%d%d%d",&n,&m,&m2);
        for(int i=1;i<=n;i++)G[i].clear(),W[i].clear();
        while(m--){
            scanf("%d%d%d",&u,&v,&w);
            G[u].push_back(v);W[u].push_back(w);
            G[v].push_back(u);W[v].push_back(w);
        }
        memset(dis,0,sizeof(dis));memset(V,0,sizeof(V));flag=0;
        while(m2--){
            scanf("%d%d%d",&u,&v,&w);
            G[u].push_back(v);W[u].push_back(-w);
            dis[v]=min(dis[v],-w);
        }
        for(int i=1;i<=n;i++)if(dis[i]<0){
            if(DFS(i)){flag=1;break;}
        }
        puts(flag?"YES":"NO");
    }
    return 0;
}

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int c,hd[600],to[6000],wt[6000],nxt[6000],dis[600],V[600],flag;
void Adde(int u,int v,int w){to[++c]=v;wt[c]=w;nxt[c]=hd[u];hd[u]=c;}
bool DFS(int u){
    V[u]=1;
    for(int i=hd[u];i;i=nxt[i]){
        int v=to[i];
        if(dis[u]+wt[i]<dis[v]){
            dis[v]=dis[u]+wt[i];
            if(V[v]==1||DFS(v))return true;
        }
    }
    V[u]=0;
    return false;
}

int main(){
    int _;scanf("%d",&_);
    while(_--){
        int n,m,m2,u,v,w;
        scanf("%d%d%d",&n,&m,&m2);
        c=0;for(int i=1;i<=n;i++)hd[i]=0;
        while(m--){scanf("%d%d%d",&u,&v,&w);Adde(u,v,w);Adde(v,u,w);}
        memset(dis,0,sizeof(dis));memset(V,0,sizeof(V));flag=0;
        while(m2--){
            scanf("%d%d%d",&u,&v,&w);Adde(u,v,-w);
            dis[v]=min(dis[v],-w);
        }
        for(int i=1;i<=n;i++)if(dis[i]<0){
            if(DFS(i)){flag=1;break;}
        }
        puts(flag?"YES":"NO");
    }
    return 0;
}