POJ-3259-Wormholes

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;
}