SPFA判环模板

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

#define INF 0x3f3f3f3f;
int t,n,m,k,a,b,d;
int mp[505][505];
int vis[505];
int cnt[505];
int dis[505];

void init(){
	for(int i=1;i<=n;i++){
		cnt[i]=0;
		vis[i]=0;
		dis[i]=INF;
		for(int j=1;j<=n;j++){
			mp[i][j]=INF;
		}
	}
}

int spfa(){
	queue<int> Q;
	vis[1]=1;
	cnt[1]=1;
	dis[1]=0;
	Q.push(1);
	while(!Q.empty()){
		int now=Q.front();Q.pop();
		vis[now]=0;
		for(int i=1;i<=n;i++){
			if(dis[i]>dis[now]+mp[now][i]){
				dis[i]=dis[now]+mp[now][i];
				if(!vis[i]){
					Q.push(i);
					vis[i]=1;
					cnt[i]++;   //在里面加 
					if(cnt[i]>=n) return 1;
				}
			}
		}
	}
	return 0;
}

int main(){
	cin>>t;
	while(t--){
		cin>>n>>m>>k;
		init();
		for(int i=0;i<m;i++){
			cin>>a>>b>>d;
			mp[a][b]=mp[b][a]=min(d,mp[a][b]);
		}
		for(int i=0;i<k;i++){
			cin>>a>>b>>d;
			mp[a][b]=min(-d,mp[a][b]);  //负权边是单向的，认真审题 
		}
		if(spfa()) cout << "YES" <<endl;
		else cout << "NO" <<endl;
	}
	
	return 0;
}