P3385 【模板】负环

点击查看代码

#include<bits/stdc++.h>
using namespace std;

const int N=2e3+10,M=6e3+10;
int t;
int n,m;

int h[N],ne[M],w[M],e[M],idx;
bool st[N];
int cnt[N];
int dist[N]; 

void add(int u,int v,int c)
{
    w[idx]=c,ne[idx]=h[u],e[idx]=v,h[u]=idx++;
}

bool spfa()
{
    memset(dist,0x3f,sizeof dist);
    memset(st,false,sizeof st);
    memset(cnt,0,sizeof cnt);

	queue<int> q;
    dist[1]=0;
    q.push(1);
    st[1]=true;

    while(q.size()){
        int t=q.front();
        q.pop();
        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]){
                    q.push(j);
                    st[j]=true;
                }
            }
        }
    }

    return false;
}

void solve()
{
    cin>>n>>m;
    while(m--){
        int u,v,w;
        cin>>u>>v>>w;
        if(w>=0){
            add(u,v,w);
            add(v,u,w);
        }else{
        	add(u,v,w);
		}
    }
    if(spfa()){
        cout<<"YES"<<endl;
    }else{
        cout<<"NO"<<endl;
    }
}

int main()
{
    ios::sync_with_stdio(0),cin.tie(0);
    

    int t;
    cin>>t;
    while(t--){
    	memset(h,-1,sizeof h);
        idx=0;
        solve();
    }

    return 0;
}