BZOJ 1715 [Usaco2006 Dec]Wormholes 虫洞 SPFA

Description

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

Input

* 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秒前的虫洞。

Output

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

Sample Input

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

Sample Output

NO
YES

HINT

传送门

来水一水……

SPFA……

#include<bits/stdc++.h>
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0' || ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int 
	maxn=505,
	inf=1000000000;
int n,m,w;
int Map[maxn][maxn],Q[maxn*maxn];
int dis[maxn],p[maxn];
void solve(){
    int x,y,z;
    n=read(),m=read(),w=read();
    
    for (int i=1;i<=n;i++)
		for (int j=1;j<=n;j++) Map[i][j]=inf;
	for (int i=0;i<=n;i++) dis[i]=inf,p[i]=0;
		
    for (int i=1;i<=m;i++){
        x=read(),y=read(),z=read();
        if (Map[x][y]>z) Map[x][y]=Map[y][x]=z;
    }
    for (int i=1;i<=w;i++){
        x=read(),y=read(),z=read();
        if (Map[x][y]>-z) Map[x][y]=-z;
    }
    int head=0,tail=1;
    Q[1]=1,dis[1]=0;
    while (head<tail){
        int u=Q[++head];
        for (int j=1;j<=n;j++)
            if (dis[j]>dis[u]+Map[u][j]){
                dis[j]=dis[u]+Map[u][j];
                Q[++tail]=j,p[j]++;
                if (p[j]==n){puts("YES");return;}
            }
        if (dis[1]<0){puts("YES");return;}
    }
    puts("NO");
}
int main(){
    int cas=read();
    while (cas--) solve();
    return 0;
}