一、导入(“虫洞”~wormhole)
//由于不知道起点,就将初始距离都设置为0,将所有点都入队。等同于加入一个超级源点 从每个点出发若能找到一条负环
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=510,M=6010;
int _,n,m,k,tot,to[M],w[M],nxt[M],h[N],d[N],cnt[N],inQ[N];
void add_edge(int a,int b,int c){
to[++tot]=b;w[tot]=c;nxt[tot]=h[a];h[a]=tot;
}
int SPFA(){
memset(d,0,sizeof(d));
memset(cnt,0,sizeof(cnt));
memset(inQ,0,sizeof(inQ));
queue<int>Q;
for(int i=1;i<=n;i++){Q.push(i);inQ[i]=1;}
while(Q.size()){
int u=Q.front();Q.pop();inQ[u]=0;
for(int i=h[u];~i;i=nxt[i]){
int v=to[i];
if(d[u]+w[i]<d[v]){
d[v]=d[u]+w[i];
if(!inQ[v]){
if(++cnt[v]>=n-1)return 1;
Q.push(v);inQ[v]=1;
}
}
}
}
return 0;
}
int main(){
scanf("%d",&_);
while(_--){
memset(h,-1,sizeof(h));tot=0;
scanf("%d%d%d",&n,&m,&k);
int u,v,w;
while(m--){
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,w);add_edge(v,u,w);
}
while(k--){
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,-w);
}
puts(SPFA()?"YES":"NO");
}
return 0;
}
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=510,M=6010;
int _,n,m,k,tot,to[M],w[M],nxt[M],h[N],d[N],cnt[N],inQ[N];
void add_edge(int a,int b,int c){
to[++tot]=b;w[tot]=c;nxt[tot]=h[a];h[a]=tot;
}
int SPFA(int s){
memset(d,0x3f,sizeof(d));
memset(cnt,0,sizeof(cnt));
memset(inQ,0,sizeof(inQ));
queue<int>Q;
Q.push(s);inQ[s]=1;cnt[s]=1;d[s]=0;
while(Q.size()){
int u=Q.front();Q.pop();inQ[u]=0;
for(int i=h[u];~i;i=nxt[i]){
int v=to[i];
if(d[u]+w[i]<d[v]){
d[v]=d[u]+w[i];
if(!inQ[v]){
if(++cnt[v]>=n)return 1;
Q.push(v);inQ[v]=1;
}
}
}
}
return 0;
}
int main(){
scanf("%d",&_);
while(_--){
memset(h,-1,sizeof(h));tot=0;
scanf("%d%d%d",&n,&m,&k);
int u,v,w;
while(m--){
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,w);add_edge(v,u,w);
}
while(k--){
scanf("%d%d%d",&u,&v,&w);
add_edge(u,v,-w);
}
puts(SPFA(1)?"YES":"NO");
}
return 0;
}
[方法1]经典方法 169ms
#include<cstdio>
#include<cstring>
#include<utility>
#include<vector>
#include<queue>
#define N 505
using namespace std;
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
vector<pair<int,int> > G[N];
int T,n,m,k,d[N],cnt[N]; bool inQ[N];
int SPFA(int s){
memset(d,0x3f,sizeof(d));
memset(cnt,0,sizeof(cnt));
memset(inQ,false,sizeof(inQ));
queue<int> Q;
Q.push(s); inQ[s]=1; cnt[s]=1; d[s]=0;
while(!Q.empty()){
int u=Q.front(); Q.pop(); inQ[u]=0;
for(int i=0;i<G[u].size();i++){
int v=G[u][i].first,w=G[u][i].second;
if(d[u]+w<d[v]){
d[v]=d[u]+w;
if(!inQ[v]){
Q.push(v); inQ[v]=1; cnt[v]++;
if(cnt[v]>n-1)return 1;
}
}
}
}
return 0;
}
int main(){
T=read();
while(T--){
for(int i=0;i<N;i++)G[i].clear();
n=read(),m=read(),k=read();
int u,v,w;
while(m--)
u=read(),v=read(),w=read(),
G[u].push_back(make_pair(v,w)),G[v].push_back(make_pair(u,w));
while(k--)
u=read(),v=read(),w=read(),
G[u].push_back(make_pair(v,-w));
puts(SPFA(1)?"YES":"NO");
}
return 0;
}
[方法2]32ms
#include<bits/stdc++.h>
using namespace std;
int dis[600],V[600],flag;
vector<int> G[600],W[600];
bool DFS(int u){
V[u]=1;
for(int i=0;i<G[u].size();i++){
int v=G[u][i];
if(dis[u]+W[u][i]<dis[v]){
dis[v]=dis[u]+W[u][i];
if(V[v]==1||DFS(v))return true;
}
}
V[u]=0;
return false;
}
int main(){
int _;scanf("%d",&_);
while(_--){
int n,m,m2,u,v,w;
scanf("%d%d%d",&n,&m,&m2);
for(int i=1;i<=n;i++)G[i].clear(),W[i].clear();
while(m--){
scanf("%d%d%d",&u,&v,&w);
G[u].push_back(v);W[u].push_back(w);
G[v].push_back(u);W[v].push_back(w);
}
memset(dis,0,sizeof(dis));memset(V,0,sizeof(V));flag=0;
while(m2--){
scanf("%d%d%d",&u,&v,&w);
G[u].push_back(v);W[u].push_back(-w);
dis[v]=min(dis[v],-w);
}
for(int i=1;i<=n;i++)if(dis[i]<0){
if(DFS(i)){flag=1;break;}
}
puts(flag?"YES":"NO");
}
return 0;
}
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int c,hd[600],to[6000],wt[6000],nxt[6000],dis[600],V[600],flag;
void Adde(int u,int v,int w){to[++c]=v;wt[c]=w;nxt[c]=hd[u];hd[u]=c;}
bool DFS(int u){
V[u]=1;
for(int i=hd[u];i;i=nxt[i]){
int v=to[i];
if(dis[u]+wt[i]<dis[v]){
dis[v]=dis[u]+wt[i];
if(V[v]==1||DFS(v))return true;
}
}
V[u]=0;
return false;
}
int main(){
int _;scanf("%d",&_);
while(_--){
int n,m,m2,u,v,w;
scanf("%d%d%d",&n,&m,&m2);
c=0;for(int i=1;i<=n;i++)hd[i]=0;
while(m--){scanf("%d%d%d",&u,&v,&w);Adde(u,v,w);Adde(v,u,w);}
memset(dis,0,sizeof(dis));memset(V,0,sizeof(V));flag=0;
while(m2--){
scanf("%d%d%d",&u,&v,&w);Adde(u,v,-w);
dis[v]=min(dis[v],-w);
}
for(int i=1;i<=n;i++)if(dis[i]<0){
if(DFS(i)){flag=1;break;}
}
puts(flag?"YES":"NO");
}
return 0;
}