1. 程式人生 > >Luogu P3385 【模板】負環

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