852. spfa判断负环

给定一个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 <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;

typedef pair<int, int> PII;
const int maxn = 1e5+5;
int h[maxn]; int ne[maxn]; int e[maxn]; int w[maxn]; int idx;
int dist[maxn];
int n, m;
bool st[maxn];
int cnt[maxn];//cnt来记录路径经过了几条边
void add(int a,int b,int c) {
	e[idx] = b;
	ne[idx] = h[a];
	w[idx] = c;
	h[a] = idx++;
}
bool spfa() {
	dist[1] = 0;
	queue<int> q;
	for(int i=1;i<=n;i++){//有几个点位起点判断是否有负环，就要把那几个点入队
	    st[i]=true;
	    q.push(i);
	}
	while(q.size()){
	    int t=q.front();
	    q.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];
	            cnt[j]=cnt[t]+1;
	            if(cnt[j]>n){//如果发现大于n了，就表示有负环
	                return true;
	            }
	            if(!st[j]){
	                q.push(j);
	                st[j]=true;
	            }
	        }
	    }
	    }
	return false;
}
int main() {
	cin >> n >> m;
	memset(dist, 0x3f, sizeof(dist));
	memset(h, -1, sizeof(h));
	while (m--) {
		int a, b,k;
		cin >> a >> b >> k;
		add(a, b, k);//这里不用处理重变
	}
	if(spfa()) puts("Yes");
	else puts("No");
	return 0;
}