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;}