pku 3259 Wormholes

#include<stdio.h>
struct node
{
int a,b,t;
}s[6000];
int dist[505],n,m,w;
int relax(int u,int v,int t)
{
if(dist[v]>dist[u]+t)
{
dist[v]=dist[u]+t;
return 1;
}
return 0;
}
int bellman(int v0)
{
int i,j;
for(i=1;i<=n;i++)
dist[i]=0xfffffff;
dist[v0]=0;
int flag;
for(i=1;i<n;i++)
{
flag=0;
for(j=1;j<=2*m+w;j++)
{
if(relax(s[j].a,s[j].b,s[j].t))
{
flag=1;
}
}
if(!flag)
break;
}
for(j=1;j<=2*m+w;j++)
if(relax(s[j].a,s[j].b,s[j].t))
return 1;
return 0;
}
int main()
{
int f,i,a,b,t;
scanf("%d",&f);
while(f--)
{
scanf("%d%d%d",&n,&m,&w);
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&t);
s[i].a=a;
s[i].b=b;
s[i].t=t;
s[i+m].a=b;
s[i+m].b=a;
s[i+m].t=t;
}
for(i=1;i<=w;i++)
{
scanf("%d%d%d",&a,&b,&t);
s[m*2+i].a=a;
s[m*2+i].b=b;
s[m*2+i].t=-t;
}
if(bellman(1))
printf("YES\n");
else
printf("NO\n");
}
return 0;
}