【最短路】P3385 【模板】负环

spfa判负环，记录每个点用于更新的次数，若该次数大于等于n次，就意味着出现了负环

 

#include<bits/stdc++.h>
using namespace std;
int t,n,m,cnt,flag;
const int inf=0x3f3f3f3f;
int head[2005],update[2005],vis[2005],dis[2005];
struct edge
{
	int to,v,nxt;
}G[6005];
void add(int x,int y,int z)
{
	G[++cnt].to=y; G[cnt].nxt=head[x]; G[cnt].v=z; head[x]=cnt;
}
void spfa()
{
	queue <int> q;
	q.push(1); dis[1]=0; 
	vis[1]=1; update[1]++; 
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		vis[u]=0;
		for(int i=head[u];i;i=G[i].nxt)
		{
			int to=G[i].to;
			int v=G[i].v;
			if(dis[to]>dis[u]+v)
			{
				update[to]++;
				dis[to]=dis[u]+v;
				if(update[to]>=n)
				{
					flag=1;
					return;
				}
				if(!vis[to])
				{
					vis[to]=1;
					q.push(to);
				}
			}
		}
	}
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);
	scanf("%d",&t);
	int u,v,w;
	while(t--)
	{
		memset(head,0,sizeof(head));
		memset(update,0,sizeof(update));
		memset(vis,0,sizeof(vis));
		memset(dis,inf,sizeof(dis));
		cnt=0;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			if(w>=0) add(u,v,w),add(v,u,w);
			else add(u,v,w);
		}
		flag=0;
		spfa();
		if(!flag) printf("NO\n");
		else printf("YES\n");
	}
	return 0;
}