P3385 模板负环

Posted 2021-11-09 辉小歌

https://www.luogu.com.cn/problem/P3385
这个和普通的判负环有点区别。普通的判负环是判断有没有负环即可。
这个的判负环是从1出发的负环。
故得将距离初始化为无穷，然后只让1入队

#include<bits/stdc++.h>
using namespace std;
const int N=1e5*2+10;
int st[N],cnt[N],dist[N],t,n,m,p[N];
int h[N],e[N],w[N],ne[N],idx;
queue<int>q,temp;
void init()
{
    q=temp;
	idx=0;
	memset(h,-1,sizeof h);
	memset(st,0,sizeof st);
	memset(cnt,0,sizeof cnt);
	memset(dist,0x3f,sizeof dist);
	for(int i=1;i<=n;i++) p[i]=i;
}
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
void add(int a,int b,int c) {e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;}
bool spfa()
{
    q.push(1),st[1]=1,dist[1]=0;
	while(q.size())
	{
		int u=q.front(); q.pop();
		st[u]=false;
		for(int i=h[u];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dist[u]+w[i])
			{
				dist[j]=dist[u]+w[i];
				cnt[j]=cnt[u]+1;
				if(cnt[j]>=n) return true;
				if(!st[j]) q.push(j),st[j]=1;
			}
		}
	}
	return false;
}
int main(void)
{
    cin>>t;
	while(t--)
	{
		cin>>n>>m;
		init();
		for(int i=0;i<m;i++)
		{
			int a,b,c; cin>>a>>b>>c;
			if(c>=0) add(a,b,c),add(b,a,c);
			else add(a,b,c);
		}
		if(spfa()) puts("YES");
		else puts("NO");
	}
	return 0;
}