Luogu P3385 【模板】负环

https://www.luogu.org/problemnew/show/P3385

题目描述

暴力枚举/SPFA/Bellman-ford/奇怪的贪心/超神搜索

输入输出格式

输入格式：

 

第一行一个正整数T表示数据组数，对于每组数据：

第一行两个正整数N M，表示图有N个顶点，M条边

接下来M行，每行三个整数a b w，表示a->b有一条权值为w的边（若w<0则为单向，否则双向）

 

输出格式：

 

共T行。对于每组数据，存在负环则输出一行"YE5"(不含引号)，否则输出一行"N0"(不含引号)。

 

输入输出样例

输入样例#1： 复制

2
3 4
1 2 2
1 3 4
2 3 1
3 1 -3
3 3
1 2 3
2 3 4
3 1 -8

输出样例#1： 复制

N0
YE5

说明

n\leq 2000n≤2000m\leq 3000m≤3000-10000\leq w\leq 10000−10000≤w≤10000T\leq 10T≤10建议复制输出格式中的字符串。 本题数据感谢@negiizhao的精心构造，请不要使用玄学算法 本题数据有更新

 

SPFA求负环，当一个点经过次数==n时跳出

#include<cstdio>
using namespace std;

const int N=1e6+5;
int n,m,l,r,d[N],num[N],q[N];
int cnt,he[N],to[N],nxt[N],w[N];
bool f,fl[N];

int read()
{
	int ret=0; bool f=0;
	char ch=getchar();
	while(ch<'0'||ch>'9') 
	{
		if(ch=='-') f=1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') 
		ret=(ret<<1)+(ret<<3)+ch-'0',ch=getchar();
	return f?-ret:ret;
}

inline void add(int u,int v,int k)
{
	to[++cnt]=v,nxt[cnt]=he[u],he[u]=cnt,w[cnt]=k;
}

int main()
{
	int T=read();
	while(T--)
	{
		n=read(),m=read();
		cnt=0;
		for(int i=1;i<=n;i++) 
			he[i]=num[i]=fl[i]=0,d[i]=1e9;
		for(int i=1;i<=m;i++)
		{
			int u=read(),v=read(),k=read();
			add(u,v,k);
			if(k>=0) add(v,u,k);
		}
		f=0;
		l=1,r=2,q[1]=1; d[1]=0;
		while(l!=r)
		{
			int u=q[l]; fl[u]=0;
			l++; if(l>n) l=0;
			for(int e=he[u];e;e=nxt[e])
			{
				int v=to[e];
				if(d[v]>d[u]+w[e])
				{
					d[v]=d[u]+w[e];
					if(!fl[v]) 
					{
						q[r]=v,r++,fl[v]=1;
						if(r>n) r=0;
					}
					num[v]++;
					if(num[v]==n) 
					{
						puts("YE5");  f=1;
						break;
					}
				}
			}
			if(f) break;
		}
		if(!f) puts("NO");
	}
	return 0;
}