思路:用bellman建立无向边,然后将虫洞的边置为负值求最短路,在进行n-1次松弛后,判断是否还能继续松弛。如果可以,则有负权回路,则可以在出发前回到原地,输出YES,否则,NO;
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define N 6000//建立无向边时注意N的范围相应的扩大为二倍
int u[N],v[N],w[N],dis[N];
int n,m,W;
int inf=9999999;
void bellman()
{
int i,j;
for(i=1; i<=n; i++)
dis[i]=inf;
dis[1]=0;
for(i=1; i<n; i++)
for(j=1; j<=2*m+W; j++)
{
if(dis[u[j]]<inf&&dis[v[j]]>dis[u[j]]+w[j])
dis[v[j]]=dis[u[j]]+w[j];
}
}
int main()
{
int t,c,a,b,i;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&W);
for(i=1; i<=2*m; i+=2)//建立无向边,正着反着各建一条边
{
scanf("%d%d%d",&a,&b,&w[i]);
u[i]=a;v[i]=b;
v[i+1]=a,u[i+1]=b;w[i+1]=w[i];
}
for(i=2*m+1; i<=W+2*m; i++)
{
scanf("%d%d%d",&u[i],&v[i],&c);
w[i]=-c;//由虫洞回去的边置为负值
}
bellman();
int flag=0;
for(i=1; i<=2*m+W; i++)
{
if(dis[u[i]]<inf&&dis[v[i]]>dis[u[i]]+w[i])
flag=1;
}
if(flag==1)
printf("YES\n");
else
printf("NO\n");
}
return 0;
}
建立无向边时候一定要注意N的范围相应的扩大!!