AcWing 904. 虫洞

题目

农夫约翰在巡视他的众多农场时，发现了很多令人惊叹的虫洞。

虫洞非常奇特，它可以看作是一条 单向 路径，通过它可以使你回到过去的某个时刻（相对于你进入虫洞之前）。

农夫约翰的每个农场中包含N片田地，M条路径（双向）以及W个虫洞。

现在农夫约翰希望能够从农场中的某片田地出发，经过一些路径和虫洞回到过去，并在他的出发时刻之前赶到他的出发地。

他希望能够看到出发之前的自己。

请你判断一下约翰能否做到这一点。

下面我们将给你提供约翰拥有的农场数量F，以及每个农场的完整信息。

已知走过任何一条路径所花费的时间都不超过10000秒，任何虫洞将他带回的时间都不会超过10000秒。

输入格式

第一行包含整数F，表示约翰共有F个农场。

对于每个农场，第一行包含三个整数N，M，W。

接下来M行，每行包含三个整数S，E，T，表示田地S和E之间存在一条路径，经过这条路径所花的时间为T。

再接下来W行，每行包含三个整数S，E，T，表示存在一条从田地S走到田地E的虫洞，走过这条虫洞，可以回到T秒之间。

输出格式

输出共F行，每行输出一个结果。

如果约翰能够在出发时刻之前回到出发地，则输出“YES”，否则输出“NO”。

数据范围

1≤F≤5
1≤N≤500,
1≤M≤2500,
1≤W≤200,
1≤T≤10000,
1≤S,E≤N

思路

判负环

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
const int N=510,M=5210;
 
int n,m1,m2;
int h[N],e[M],ne[M],w[M],idx;
int dist[N];
int q[N],cnt[N];
bool st[N];

void add(int a,int b,int c)
{
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

bool spfa()
{
	int hh=0,tt=0;
	memset(dist,0,sizeof dist);
	memset(st,0,sizeof st);
	memset(cnt,0,sizeof cnt);
	
	for(int i=1;i<=n;i++)
	{
		q[tt++]=i;
		st[i]=true; 
	} 
	while(hh!=tt)
	{
		int t=q[hh++];
		if(hh==N) hh=0;
		st[t]=false;
		for(int i=h[t];~i;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dist[t]+w[i])
			{
				dist[j]=dist[t]+w[i];
				cnt[j]=cnt[t]+1;
				if(cnt[j]>=n) return true;
				if(!st[j])
				{
					q[tt++]=j;
					if(tt==N) tt=0;
					st[j]=true;
				}
			}
		}
	}
	return false;
}

int main()
{
	//freopen("test.in","r",stdin);//设置 cin scanf 这些输入流都从 test.in中读取
    //freopen("test.out","w",stdout);//设置 cout printf 这些输出流都输出到 test.out里面去
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n>>m1>>m2;
		memset(h,-1,sizeof h);
		idx = 0;
		while(m1--)
		{
			int a,b,c;
			cin>>a>>b>>c;
			add(a,b,c),add(b,a,c);
		}
		while(m2--)
		{
			int a,b,c;
			cin>>a>>b>>c;
			add(a,b,-c);
		}
		if(spfa()) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	}
	return 0;
}