「一本通 3.3 练习 2」虫洞 Wormholes （SPFA 负环）

哆啦B梦的传送门

分析：把可以穿越的路径放入邻接表中，并且将距离改为负值，直接套入判断负环的模板，如果是负环，肯定可以穿越回去（一次时间不够，可以走多次负环）。
AC代码：

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
int num,n,flag;
const int maxn=1e6+5;
int vis[maxn],sum[maxn],head[maxn],dis[maxn];
int inf=0x3f3f3f3f;
struct node
{
	int to,nex,w;
}l[maxn];
void add(int u,int v,int w)
{
	num++;
	l[num].to=v;
	l[num].w=w;
	l[num].nex=head[u];
	head[u]=num;
}
int spfa()
{
	int i,h,t;
	queue<int> q;
	for(i=1;i<=n;i++)
	dis[i]=inf;
	dis[1]=0;
	vis[1]=1;
	sum[1]++; 
	q.push(1);
	while(!q.empty())
	{
		h=q.front();
		q.pop();
		vis[h]=0;
		for(i=head[h];i!=-1;i=l[i].nex)
		{
			t=l[i].to;
			if(dis[h]+l[i].w<dis[t])
			{
				dis[t]=dis[h]+l[i].w;
				sum[t]++;
				if(sum[t]>n)//如果不是负环，最多可以遍历n-1次，因为有n个点 
				return 1; 
				if(!vis[t])
				{
					vis[t]=1;
					q.push(t);
				}
			}
		}
	}
	return 0;
}
int main()
{
	int t,s1,s2,u,v,w;
	scanf("%d",&t);
	while(t--)
	{
	flag=0;
	num=0;
	memset(head,-1,sizeof(head));
	memset(vis,0,sizeof(vis));
	memset(sum,0,sizeof(sum));
	scanf("%d%d%d",&n,&s1,&s2);
	while(s1--)
	{
	scanf("%d%d%d",&u,&v,&w);//无向边 
	add(u,v,w);
	add(v,u,w);
	}
	while(s2--)
	{
	scanf("%d%d%d",&u,&v,&w);//有向边 
	add(u,v,-w);	
	}
	if(spfa())
	printf("YES\n");
	else printf("NO\n");
	}
}