Wormholes POJ 3259

用Bellman-ford算法看这张图有没有负权回路。

还不是很懂要多看看

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int MAXN=10050;
const int INF=0x3f3f3f3f;

struct node{
	int from;
	int to;
	int cost;
}map[MAXN];

int dis[MAXN];
int n,m,k;
int num;

void add(int x,int y,int z)
{
	map[num].from=x;
	map[num].to=y;
	map[num].cost=z;
	num++;
}

int shoetest(int s)
{
	for(int i=1;i<=n;i++)
		dis[i]=INF;
	dis[s]=0;
	for(int i=0;i<n;i++)
	{
		bool update=false;
		for(int i=0;i<num;i++)
		{
			node e=map[i];
			if(dis[e.to]>dis[e.from]+e.cost)
			{
				dis[e.to]=dis[e.from]+e.cost;
				update=true;
			}
		}
		if(!update)
			break;
	}
	for(int i=0;i<num;i++)
		if(dis[map[i].to]>dis[map[i].from]+map[i].cost)
			return 1;
	return 0;
}

int main()
{
	int t;
	int x,y,z;
	cin>>t;
	while(t--)
	{
		cin>>n>>m>>k;
		num=0;
		for(int i=0;i<m;i++)
		{
			cin>>x>>y>>z;
			add(x,y,z);
			add(y,x,z);
		}
		for(int i=0;i<k;i++)
		{
			cin>>x>>y>>z;
			add(x,y,-z);
		}
		if(shoetest(0))
			cout<<"YES"<<endl;
		else
			cout<<"NO"<<endl;
	}
	return 0;
}