洛谷 P3385 【模板】负环

给一个 $n$ 点有向图，从 $1$ 出发，问是否能到负环。

思路：SPFA 判负环，假如入队 $n$ 次以上，则存在负环。

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#define MAXN 2010
#define MAXM 6010
#define INF 0x3f3f3f3f
using namespace std;
int n,m,T,head[MAXN],d[MAXN],cnt,u,v,w,in[MAXN];
bool inq[MAXN];
struct Edge{
    int to,nxt,w;
}edge[MAXM];
void addedge(int u,int v,int w){
    edge[cnt].to=v;edge[cnt].nxt=head[u];
    edge[cnt].w=w;head[u]=cnt++;
}
bool spfa(){
    queue<int> q;
    q.push(1);
    memset(inq,false,sizeof(inq));
    memset(in,0,sizeof(in));
    in[1]++;
    inq[1]=true;
    memset(d,INF,sizeof(d));
    d[1]=0;
    while(!q.empty()){
        int u=q.front();q.pop();
        inq[u]=false;
        for(int i=head[u];i!=-1;i=edge[i].nxt){
            int v=edge[i].to,w=edge[i].w;
            if(d[u]+w<d[v]){
                d[v]=d[u]+w;
                if(!inq[v]){
                    q.push(v);
                    in[v]++;
                    if(in[v]>=n)return true;
                }
            }
        }
    }
    return false;
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf("%d",&T);
    while(T--){
        memset(head,-1,sizeof(head));
        cnt=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
            if(w>=0)addedge(v,u,w);
        }
        if(spfa())printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}