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\n");
else printf("NO\n");
}

return 0;
}