AcWing 852. spfa判断负环(spfa)

#include<bits/stdc++.h>
using namespace std;

const int N=100010;
int n,m;
int h[N],e[N],ne[N],w[N],idx;
int dist[N],cnt[N];
bool st[N];
queue<int> que;

void add(int x,int y,int z){
	e[idx]=y; w[idx]=z; ne[idx]=h[x];
	h[x]=idx;
	idx++;
	return ;
}

bool spfa(){
	memset(dist,0x3f,sizeof(dist));
	dist[1]=0;
	for(int i=1;i<=n;i++){
	    que.push(i);
	}
	st[1]=true;
	while(!que.empty()){
		auto t=que.front();
		que.pop();
		st[t]=false;
		for(int i=h[t];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>dist[t]+w[i]){
				dist[j]=dist[t]+w[i];
				que.push(j);
				cnt[j]=cnt[t]+1;
				if(cnt[j]>=n){
					return true;
				}
				if(!st[j]){
					st[j]=true;
				}
			}
		}
	}
	return false;
}

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n>>m;
	memset(h,-1,sizeof(h));
	for(int i=1;i<=m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		add(x,y,z);
	}
	if(spfa()){
		cout<<"Yes";
	}
	else{
		cout<<"No";
	}
	return 0;
}