传送门
Wormholes
题意:给出一张图,判断是否存在负环。
I think
方法:SPFA求单源最短路,若一个点入队超过N次,那么存在负环。
Code
#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;
const int sm = 5250;
const int sn = 505;
const int Inf = 6e6;
int T,N,M,W,tot;
int hd[sn],to[sm],nxt[sm],w[sm];
int cnt[sn],dis[sn];
bool vis[sn];
queue<int>Q;
void Add(int u,int v,int c) {
to[++tot]=v,nxt[tot]=hd[u],hd[u]=tot,w[tot]=c;
}
bool SPFA() {
while(!Q.empty()) Q.pop();
for(int i=1;i<=N;++i)
dis[i]=Inf,cnt[i]=0,vis[i]=0;
int t; dis[1]=0;Q.push(1);
vis[1]=1;++cnt[1];
while(!Q.empty()) {
t=Q.front(),Q.pop();
vis[t]=false;
for(int i=hd[t];i;i=nxt[i])
if(dis[to[i]]>dis[t]+w[i]) {
dis[to[i]]=dis[t]+w[i];
if(!vis[to[i]]) {
Q.push(to[i]),++cnt[to[i]];
vis[to[i]]=true;
if(cnt[to[i]]>N) return true;
}
}
}
return false;
}
void init() {
tot=0;
memset(hd,0,sizeof(hd));
}
int main() {
int u,v,w;
scanf("%d",&T);
while(T--) {
init();
scanf("%d%d%d",&N,&M,&W);
for(int i=1;i<=M;++i) {
scanf("%d%d%d",&u,&v,&w);
Add(u,v,w),Add(v,u,w);
}
for(int i=1;i<=W;++i) {
scanf("%d%d%d",&u,&v,&w);
Add(u,v,-w);
}
puts(SPFA()?"YES":"NO");
}
return 0;
}