POJ-3259-Wormholes
http://poj.org/problem?id=3259
bellman_ford判断是否有负环
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
#define MAX 0x7fffffff
struct edge
{
int x,y,v;
}edge[6000];
int m,n,w;
int dis[600];
int bellman_ford(int k)
{
int i,j,flag;
for(i=1;i<=n;i++)
dis[i]=MAX;
dis[1]=0;
flag=0;
for(i=1;i<n;i++) //松弛
{
flag=0;
for(j=0;j<k;j++)
if(dis[edge[j].x]!=MAX&&dis[edge[j].y]>dis[edge[j].x]+edge[j].v)
{
dis[edge[j].y]=dis[edge[j].x]+edge[j].v;
flag=1;
}
if(!flag) //一次更新之后没有变化,下次就不用重复更新
break;
}
for(i=0;i<k;i++) //加入这条边仍能使路劲变短,说明有负环
if(dis[edge[i].y]>dis[edge[i].x]+edge[i].v)
return 1;
return 0;
}
int main()
{
int t;
int a,b,c,k;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&w);
k=0;
while(m--)
{
scanf("%d%d%d",&a,&b,&c);
edge[k].x=a;
edge[k].y=b;
edge[k].v=c;
k++;
edge[k].x=b;
edge[k].y=a;
edge[k].v=c;
k++;
}
while(w--)
{
scanf("%d%d%d",&a,&b,&c);
edge[k].x=a;
edge[k].y=b;
edge[k].v=-c;
k++;
}
if(bellman_ford(k))
printf("YES\n");
else
printf("NO\n");
}
return 0;
}