[poj3259] Wormholes

题目链接

题解：spfa判负环，有时间写一下dfs版spfa判环

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
#define INF 0x3f3f3f3f
const int M=10005;
int m,n,x,y,z,t,T,w,flag;
int vis[M],head[M],d[M],inq[M];
queue <int> q;

struct edge{
    int from,to,next,val;
}e[M*10];

inline void ready(int s)
{
    for(int i=1;i<=n;i++)
    d[i]=INF,vis[i]=0,inq[i]=0;
    d[s]=0;inq[s]=1;flag=0;
}

inline void add(int i,int j,int v)
{
    e[t].from=i;
    e[t].to=j;
    e[t].val=v;
    e[t].next=head[i];
    head[i]=t++;
}

bool spfa(int s)
{
    ready(s);
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();q.pop();vis[u]=0;
        for(int i=head[u];i!=-1;i=e[i].next)
        {
            int v=e[i].to;
            if(d[v]>d[u]+e[i].val)
            {
                d[v]=d[u]+e[i].val;
                if(!vis[v])
                {
                    if(++inq[v]==n) return true;
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
    return false;
}
inline void work()
{
    memset(head,-1,sizeof(head));
    scanf("%d%d%d",&n,&m,&w);
    while(m--){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    while(w--){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,-z);
    }
    for(int i=1;i<=n;i++)
//防止图不联通，可以用并查集检查一下图的连通分量
    {
        if(spfa(i))
        {
            flag=1;
            break;
        }
    } 
    if(flag)
    printf("YES\n");
    else printf("NO\n");
}
int main()
{
    cin>>T;
    while(T--){
        work();
    }
    return 0;
}