BZOJ1715: [Usaco2006 Dec]Wormholes 虫洞 spfa 求负环

此题的题意就是求图中有没有负环存在

Spfa求负环方法：当一个点已经被松弛了超过n次，则这个图存在着负环

建图的话注意：把虫洞的边的权值为数据给的权值的相反数，其它的都正常

因为题中有多组数据所以一定要记得很多数组需要清空

用spfa求负环即可。

#include<cmath>
#include<queue> 
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=505;
const int M=10005;
const int inf=2100000000;
queue<int>Q;
int n,m,W,T;
int cnt,to[M],nxt[M],w[M],pre[N];
void insert(int f,int t,int p)
{
	to[++cnt]=t;
	nxt[cnt]=pre[f];
	pre[f]=cnt;
	w[cnt]=p;
}
int d[N],cir[N];
bool inq[N];
bool spfa()
{
	memset(cir,0,sizeof(cir));
	memset(inq,0,sizeof(inq));
	for(int i=1;i<=n;i++) d[i]=inf;
	d[1]=0,cir[1]=1;
	Q.push(1);
	int h=1,t=1,x;
	while(!Q.empty())
	{
		x=Q.front();
		inq[x]=false;
		for(int i=pre[x];i;i=nxt[i])
		if(d[to[i]]>d[x]+w[i])
		{
			d[to[i]]=d[x]+w[i];
			if(++cir[to[i]]>n) return true;
			if(!inq[to[i]])
			{
				Q.push(to[i]);
				inq[to[i]]=true; 
			}
		}
		Q.pop();
	}
	return false;
}
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		cnt=0;
		memset(pre,0,sizeof(pre));
		scanf("%d%d%d",&n,&m,&W);
		int x,y,p;
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&x,&y,&p);
			insert(x,y,p),insert(y,x,p);
		}
		for(int i=1;i<=W;i++)
		{
			scanf("%d%d%d",&x,&y,&p);
			insert(x,y,-p);
		}
		if(spfa()) printf("YES\n");
		else printf("NO\n");
		
	}
}