虫洞

Problem 3 虫洞(wormhole.cpp/c/pas)

【题目描述】

John在他的农场中闲逛时发现了许多虫洞。虫洞可以看作一条十分奇特的有向边，并可以使你返回到过去的一个时刻（相对你进入虫洞之前）。John的每个农场有M条小路（无向边）连接着N （从1..N标号）块地，并有W个虫洞（有向边）。其中1<=N<=500,1<=M<=2500,1<=W<=200。 现在John想借助这些虫洞来回到过去（出发时刻之前），请你告诉他能办到吗。 John将向你提供F(1<=F<=5)个农场的地图。没有小路会耗费你超过10000秒的时间，当然也没有虫洞回帮你回到超过10000秒以前。

【输入格式】

* Line 1: 一个整数 F, 表示农场个数。

* Line 1 of each farm: 三个整数 N, M, W。

* Lines 2..M+1 of each farm: 三个数（S, E, T）。表示在标号为S的地与标号为E的地中间有一条用时T秒的小路。

* Lines M+2..M+W+1 of each farm: 三个数（S, E, T）。表示在标号为S的地与标号为E的地中间有一条可以使John到达T秒前的虫洞。

【输出格式】

* Lines 1..F: 如果John能在这个农场实现他的目标，输出"YES",否则输出"NO"。

【样例输入】

2

3 3 1

1 2 2

1 3 4

2 3 1

3 1 3

3 2 1

1 2 3

2 3 4

3 1 8

【样例输出】

NO

YES

【题解】

　　判断有无负权回路

　　这道题的数据竟然卡普通的spfa。

　　用dfs版的spfa

【代码】

#include <cstdio>
#include <algorithm>
#include <cstring>
#define ll long long
#define inf 5000010
#define maxn 600
#define maxm 6000 
using namespace std;
ll head[maxn], next[maxm], to[maxm], w[maxm], dist[maxm], n, m, F, tot, vis[maxn];
void add(ll a, ll b, ll v){to[++tot]=b;w[tot]=v;next[tot]=head[a];head[a]=tot;}
void init()
{
	ll i, j, a, b, v, c, *p;
	memset(head,0,sizeof(head));
	memset(vis,0,sizeof(vis));
	memset(dist,0x3f,sizeof(dist));
	tot=0;
	scanf("%I64d%I64d%I64d",&n,&m,&c);
	for(i=1;i<=m;i++)scanf("%I64d%I64d%I64d",&a,&b,&v),add(a,b,v),add(b,a,v);
	for(i=1;i<=c;i++)scanf("%I64d%I64d%I64d",&a,&b,&v),add(a,b,-v);
}
bool spfa(ll pos)
{
	ll p;
	vis[pos]=true;
	for(p=head[pos];p;p=next[p])
	{
		if(dist[to[p]]>dist[pos]+w[p])
		{
			dist[to[p]]=dist[pos]+w[p];
			if(vis[to[p]])return true;
			if(spfa(to[p]))return true;
		}
	}
	vis[pos]=false;
	return false;
}
int main()
{
	freopen("wormhole.in","r",stdin);
	freopen("wormhole.out","w",stdout);
	scanf("%I64d",&F);
	while(F--)
	{
		init();
		dist[1]=0;
		printf(spfa(1)?"YES\n":"NO\n");
	}
	return 0;
}