POJ 3259

求是否存在负圈环~ 把NO写成了N0，调了一下午orz~~

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define inf 0x7ffffff
int n,m,s,k;
int d[510];
struct Edge
{
  int u,v,w;
}edge[5300];
int bellmanford()
{
  int flag;
  for(int i= 1;i <= n;i++)
  d[i]=inf;
  d[1] = 0;
  for(int i = 0;i< n-1 ;i++)
  {
    flag = 0;
    for(int j = 0;j < k;j ++)
    {
      int x=edge[j].u;
      int y=edge[j].v;
      if(d[y]>d[x]+edge[j].w)
      {
          d[y] = d[x]+edge[j].w;
          flag =1;
      }
    }
    if(flag==0) break;
  }
  for(int i=0;i<k;i++)
  {
    int x = edge[i].u;
    int y = edge[i].v;
    if(d[y]>d[x]+edge[i].w)
    return 1;
  }
  return 0;
}
int main()
{
    int t,x,y,z;
    scanf("%d",&t);
    while(t--)
    {
     k=0;
     scanf("%d%d%d",&n,&m,&s);
     while(m--)
     {
      scanf("%d%d%d",&x,&y,&z);
      edge[k].u=x;
      edge[k].v=y;
      edge[k].w=z;
      k++;
      edge[k].u=y;
      edge[k].v=x;
      edge[k].w=z;
      k++;
     }
     while(s--)
     {
      scanf("%d%d%d",&x,&y,&z);
      edge[k].u=x;
      edge[k].v=y;
      edge[k].w=-z;
      k++;
     }
     int ans = bellmanford();
     if(ans==1)
     printf("YES\n");
     else
     printf("NO\n");
    }
    return 0;
}