SPFA判负环

使用spfa算法解决是否存在负环问题

求负环的常用方法，基于SPFA，一般都用方法 2（该题也是用方法 2）：

方法 1：统计每个点入队的次数，如果某个点入队n次，则说明存在负环
方法 2：统计当前每个点的最短路中所包含的边数，如果某点的最短路所包含的边数大于等于n，则也说明存在环
给定一个 n 个点 m 条边的有向图，图中可能存在重边和自环， 边权可能为负数。

请你判断图中是否存在负权回路。

输入格式
第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z，表示存在一条从点 x 到点 y 的有向边，边长为 z。

输出格式
如果图中存在负权回路，则输出 Yes，否则输出 No。

数据范围
1≤n≤2000,
1≤m≤10000,
图中涉及边长绝对值均不超过 10000。

输入样例：
3 3
1 2 -1
2 3 4
3 1 -4
输出样例：
Yes

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+100;
int head[N],v[N],ver[N],idx,nxt[N],edg[N],dis[N],cnt[N],n,m;
void add(int x,int y,int z){
	ver[++idx]=y;
	edg[idx]=z;
	nxt[idx]=head[x];
	head[x]=idx;
}
int spfa(){
	queue<int>q;
	memset(dis,0x3f,sizeof dis);
	dis[1]=0;
	cnt[1]=1;
	q.push(1);
	v[1]=1;
	for(int i=1;i<=n;i++){//可能有的图并不连通，故需要将所有的点入队，亦可理解为添加了一个虚拟点，改点到所有的点的距离都为0
		v[i]=1;
		q.push(i);
	}//洛谷P3385 【模板】负环则不同，其只需要判断是否有连接的负环，故不需要此段
	while(q.size()){
		int x=q.front();
		q.pop();
		v[x]=0;
		for(int i=head[x];i;i=nxt[i]){
			int y=ver[i],z=edg[i];
			if(dis[y]>dis[x]+z){
				dis[y]=dis[x]+z;
				cnt[y]=cnt[x]+1;//统计该点的最短路所通过的边
				if(cnt[y]>n)return 1;//若某点通过了n条边则必有负环
				if(!v[y]){
					q.push(y);
					v[y]=1; 
				}
			}
		}
	}
	return 0;
}
int main(){
	cin>>n>>m;
	while(m--){
		int x,y,z;
		scanf("%d %d %d",&x,&y,&z);
		add(x,y,z);
	}
	if(spfa())cout<<"Yes";
	else cout<<"No";
	//cout<<dis[n];
	return 0;
}