题目大意:给出n个顶点的无向图,并且在之后新增负边(虫洞)。问是否存在负环。
分析:使用spfa寻找负环。当一个节点入队列次数超过n-1时。表示存在负环。
#include <iostream>
#include <cstring>
#include <cstdio>
#define MAXN 3000
#define MAXM 3000
using namespace std;
struct Edge {
int to, len, next;
} edge[MAXM * 2];
int en, head[MAXN];
inline void add_edge(int u, int v, int len) {
edge[en].to = v, edge[en].len = len, edge[en].next = head[u], head[u] = en++;
}
int n,m,k;
int count[3000],q[1000000],f[3000];
bool vis[3000];
bool spfa()
{
int h,t,i;
memset(vis,false,sizeof(vis));
memset(count,0,sizeof(count));
memset(f,10000,sizeof(f));
h=1;t=1;q[1]=1;vis[1]=true;f[1]=0;count[1]=1;f[1]=0;
while (h<=t)
{
i=q[h];vis[i]=false;
for (int p=head[i];p!=-1;p=edge[p].next)
{
int to=edge[p].to,len=edge[p].len;
if (f[to]>f[i]+len)
{
f[to]=f[i]+len;
if (!vis[to])
{
t++;
q[t]=to;
vis[to]=true;
count[to]++;
if (count[to]>=n) return true;
}
}
}
h++;
}
return false;
}
int main()
{
int cases,u,v,len;
scanf("%d",&cases);
while (cases--)
{
scanf("%d%d%d",&n,&m,&k);
fill(head+1,head+n+1,-1);
en=0;
for (int i=1;i<=m;i++)
{
scanf("%d%d%d",&u,&v,&len);
add_edge(u,v,len);
add_edge(v,u,len);
}
for (int i=1;i<=k;i++)
{
scanf("%d%d%d",&u,&v,&len);
add_edge(u,v,-len);
}
if (spfa()) printf("YES\n");
else printf("NO\n");
}
return 0;
}