AcWing852.spfa判断负环

cnt数组表示：cnt【j】表示边j

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define N 2010
#define M 10010
using namespace std;
int n,m;
int h[N],w[M],e[M],ne[M],idx;
int dis[N],cnt[N];
bool st[N];
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;
    //题目是让判断是否存在负环，也就是判断从点1~点n判断是否存在到某个点的负环
    //因此不能仅把点1放入队列
    //把每个点都放入队列，就能正确的判断是否存在负环了，而不是是否存在点1到某个点的负环
    for(int i=1;i<=n;++i){
        st[i]=1;
        q.push(i);
    }
    while(q.size()){
        int t=q.front();
        q.pop();
        st[t]=0;
        for(int i=h[t];i!=-1;i=ne[i]){
            int j=e[i];
            if(dis[j]>dis[t]+w[i]){
                dis[j]=dis[t]+w[i];
                cnt[j]=cnt[t]+1;
                //
                if(cnt[j]>=n) return 1;
                if(!st[j]){
                    q.push(j);
                    st[j]=1;
                }
            }
        }
    }
    return false;
}
int main(){
    cin>>n>>m;
    memset(h,-1,sizeof h);
    while(m--){
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }
    puts(spfa()?"Yes":"No");
    return 0;
}