在SPFA中,每个点最多被松弛n-1遍,所以一个点被松弛n遍后可以认为出现了负环。
代码如下:
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<iostream>
using namespace std;
struct edge{int v;int w;};
vector<edge>g[10050];
int dis[6050];//从始点出发的距离
int vis[6050];
int ji[6050];//统计松弛次数
int n,m;
int t;
int st=1;
int spfa(){
queue<int> q;
vis[st]=1;
dis[st]=0;
ji[st]=1;
q.push(st);
while(!q.empty()){
int u=q.front();
q.pop();
vis[u]=0;
for(int i=0;i<g[u].size();i++){
int v=g[u][i].v,w=g[u][i].w;
if(dis[v]>dis[u]+w){
dis[v]=dis[u]+w;
if(!vis[v]){
vis[v]=1;
q.push(v);
ji[v]++;
if(ji[v]>=n)
return 1;
}
}
}
}
return 0;
}
int main(){
cin>>t;
while(t--){
cin>>n>>m;
for(int i=0;i<=n;i++){
g[i].clear();
vis[i]=0;
dis[i]=0x3f3f3f3f;
ji[i]=0;
}
for(int i=1;i<=m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
if(w>=0){
g[u].push_back((edge){v,w});
g[v].push_back((edge){u,w});
}
else
g[u].push_back((edge){v,w});
}
int ans=spfa();
if(ans)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
return 0;
}