poj 3259 Wormholes（spfa 判环）

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define INF 100000000
int u[6000],v[6000],w[6000];
int first[6000],next[6000];
int coun[6000];
__int64 d[6000];
int main()
{
    int f,n,m,ww,s,e,t;
    int i,j,k;
    scanf("%d",&f);
    while(f--)
    {
        scanf("%d%d%d",&n,&m,&ww);
        memset(first,-1,sizeof(first));
        memset(coun,0,sizeof(coun));
        int ok=0;
        for(i=1;i<=2*m;i+=2)
        {
            scanf("%d%d%d",&s,&e,&t);
            u[i]=s;  v[i]=e;  w[i]=t;
            u[i+1]=e;v[i+1]=s;w[i+1]=t;

            next[i]=first[u[i]];
            first[u[i]]=i;
            next[i+1]=first[u[i+1]];
            first[u[i+1]]=i+1;
        }
        for(i=2*m+1;i<=2*m+ww;i++)
        {
            scanf("%d%d%d",&s,&e,&t);
            u[i]=s;v[i]=e;w[i]=-t;

            next[i]=first[u[i]];
            first[u[i]]=i;
        }

        //queue<int> q;
        int tot[6000];
        int cnt=0;
        bool inq[6000];
        for(i=0;i<=n;i++) d[i]=INF;
        d[1]=0;
        memset(inq,0,sizeof(inq));
        //q.push(1);
        tot[cnt++]=1;
        coun[1]++;
        //while(!q.empty())
        while(cnt)
        {
            if(ok==1) break;
            //int x=q.front();q.pop();
            int x=tot[--cnt];
            inq[x]=false;
            for(int e=first[x];e!=-1;e=next[e])
            {
                if(d[v[e]]>d[x]+w[e])
                {
                    d[v[e]]=d[x]+w[e];
                    if(!inq[v[e]])
                    {
                        inq[v[e]]=true;
                        //q.push(v[e]);
                        tot[cnt++]=v[e];
                        coun[v[e]]++;
                        if(coun[v[e]]>n) {ok=1;break;}
                    }
                }
            }
        }
        if(ok==1) printf("YES
");
        else printf("NO
");
    }

    return 0;
}