洛谷P3385【spfa模板】判断负环

题目链接 洛谷P3385【spfa模板】判断负环
题目描述
给定一个 nn 个点的有向图，请求出图中是否存在从顶点 1 出发能到达的负环。
负环的定义是：一条边权之和为负数的回路。
输入格式
本题单测试点有多组测试数据。
输入的第一行是一个整数 TT，表示测试数据的组数。对于每组数据的格式如下：
第一行有两个整数，分别表示图的点数 nn 和接下来给出边信息的条数 mm。
接下来 mm 行，每行三个整数 u, v, wu,v,w。
若 w \geq 0w≥0，则表示存在一条从 uu 至 vv 边权为 ww 的边，还存在一条从 vv 至 uu 边权为 ww 的边。
若 w < 0w<0，则只表示存在一条从 uu 至 vv 边权为 ww 的边。
输出格式
对于每组数据，输出一行一个字符串，若所求负环存在，则输出 YES，否则输出 NO。
输入输出样例
输入 #1
2
3 4
1 2 2
1 3 4
2 3 1
3 1 -3
3 3
1 2 3
2 3 4
3 1 -8
输出 #1
NO
YES

解题代码如下

#include <bits/stdc++.h>
using namespace std;
const int N = 1e4+10;
typedef long long ll;
ll h[N],ne[N],e[N],w[N],idx;
ll dist[N];
ll cnt[N];
bool st[N];
int n,m;
typedef pair<int,int> pii;
void add(ll a,ll b,ll c){ 				///链式前向星，存储邻接表
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=c;
    h[a]=idx++;
}
void init(){			
    idx = 0;
    memset(h,-1,sizeof h);
    memset(cnt,0,sizeof cnt);
    memset(dist,0x3f,sizeof dist);
}
bool spfa(){

    queue<pii> q;
    dist[1]=0;
    q.push({0,1});st[1]=true;                               ///将当前点入队表示当前点st[i]=true表示当前点在队列中

    while(q.size()){
        pii now = q.front();q.pop();
        int t = now.second;st[t]=false;                     ///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) return true;
                if(!st[j]) st[j]=true;q.push({dist[j],j});  ///如果当前点不在队列中则将当前点加入队列
            }
        }
    }
    return false;
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        init();				///因为是多组 数据输入，故每次需将邻接表初始化
        scanf("%d%d",&n,&m);
        while(m--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
            if(c>=0) add(b,a,c);
        }
        if(spfa()) puts("YES");		///puts 自带换行
        else puts("NO");
    }
    return 0;
}