acwing-852.spfa判负环

题目链接 acwing-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 <bits/stdc++.h>
using namespace std;
const int N = 1e4+10;
int h[N],ne[N],e[N],w[N],idx;
int dist[N];
int cnt[N];
bool st[N];
int n,m;
typedef pair<int,int> pii;
void add(int a,int b,int c){ ///链式前向星，存储邻接表
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=c;
    h[a]=idx++;
}
bool spfa(){
    queue<int> q;
    for(int i=1;i<=n;i++) q.push(i),st[i]=true;
    while(q.size()){
        int now = q.front();q.pop();st[now]=false;                     ///st[t]=false表示当前点不在队列中
        for(int i=h[now];i!=-1;i=ne[i]){
            int j=e[i];
            if(dist[j]>dist[now]+w[i]){
                dist[j]=dist[now]+w[i];
                cnt[j]=cnt[now]+1;
                if(cnt[j]>=n) return true;
                if(!st[j]) st[j]=true;q.push(j);  ///如果当前点不在队列中则将当前点加入队列
            }
        }
    }
    return false;
}
int main(){
    memset(h,-1,sizeof h);
    memset(dist,0x3f,sizeof dist);
    scanf("%d%d",&n,&m);
    while(m--){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    if(spfa()) puts("Yes");
    else puts("No");
    return 0;
}