洛谷-3385 【模板】负环

题目描述
暴力枚举/SPFA/Bellman-ford/奇怪的贪心/超神搜索
寻找一个从顶点1所能到达的负环，负环定义为：一个边权之和为负的环。
输入格式
第一行一个正整数T表示数据组数，对于每组数据：
第一行两个正整数N M，表示图有N个顶点，M条边
接下来M行，每行三个整数a b w，表示a->b有一条权值为w的边（若w<0则为单向，否则双向）
输出格式
共T行。对于每组数据，存在负环则输出一行"YE5"(不含引号)，否则输出一行"N0"(不含引号)。

输入输出样例
输入 #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
N0
YE5

说明/提示
n≤2000
m≤3000
−10000≤w≤10000
T≤10

#include<bits/stdc++.h>
#define N 100086
using namespace std;
void read(int &x){
    int f=1;x=0;char s=getchar();
    while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
    while(s<='9'&&s>='0'){x=x*10+s-'0';s=getchar();}
    x*=f;
}
int n,m,T;
struct code{int u,v,w;}edge[N];
bool vis[N];
int head[N],tot,dis[N],cnt[N];
void add(int x,int y,int z){edge[++tot].u=head[x];edge[tot].v=y;edge[tot].w=z;head[x]=tot;}
bool spfa(int now){
    for(int i=1;i<=n;i++) vis[i]=false,dis[i]=2147483647,cnt[i]=false;
    queue<int>q;
    q.push(now);
    vis[now]=true;
    dis[now]=0;
    while(!q.empty()){
        int u=q.front();q.pop();vis[u]=false;
        if(cnt[u]>=n)return true;
        for(int i=head[u];i;i=edge[i].u){
            if(dis[edge[i].v]>dis[u]+edge[i].w){
                dis[edge[i].v]=dis[u]+edge[i].w;
                cnt[edge[i].v]++;
                if(cnt[edge[i].v]>=n)return true;
                if(!vis[edge[i].v]){
                    q.push(edge[i].v);
                    vis[edge[i].v]=true;
                }
            }
        }
    }
    return false;
}
int main(){
    read(T);
    while(T--){
        read(n),read(m);
        tot=0;memset(head,0,sizeof(head));
        for(int i=1,u,v,w;i<=m;i++){
            read(u),read(v),read(w);
            if(w<0)add(u,v,w);
            else add(u,v,w),add(v,u,w);
        }
        puts(spfa(1)?"YE5":"N0");
    }
}