BZOJ 1715: [Usaco2006 Dec]Wormholes 虫洞

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"。

题解：

题意是问你一个混合图是否存在负环。

spfa即可，开始时将所有点入队，求最短路，当最短路长度超过n时，说明有负环。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
//by zrt
//problem:
using namespace std;
int n,m,w;
int H[505],X[6000],P[6000],E[6000];
int d[505],l[505];
int tot;
inline void add(int x,int y,int z){
    P[++tot]=y;X[tot]=H[x];H[x]=tot;E[tot]=z;
}
int tt;
queue<int> q;
int main(){
    #ifdef LOCAL
    freopen("in.txt","r",stdin);
//  freopen("out.txt","w",stdout);
    #endif
    scanf("%d",&tt);
    while(tt--){
        memset(H,0,sizeof H);
        tot=0;
        scanf("%d%d%d",&n,&m,&w);
        for(int i=1,x,y,z;i<=m;i++){
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
        for(int i=1,x,y,z;i<=w;i++){
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,-z);
        }
        memset(d,0,sizeof d);
        memset(l,0,sizeof l);
        while(!q.empty()) q.pop();
        for(int i=1;i<=n;i++) q.push(i);
        bool have=0;
        while(!q.empty()){
            int x=q.front();q.pop();
    //      printf("%d %d %d\n",x,d[x],l[x]);
            for(int i=H[x];i;i=X[i]){
                if(d[P[i]]>d[x]+E[i]){
                    d[P[i]]=d[x]+E[i];
                    q.push(P[i]);
                    l[P[i]]=l[x]+1;
                    if(l[P[i]]>n) {
                        have=1;goto ed;
                    }
                }
            }
        }
        ed:;
        if(have) puts("YES");
        else puts("NO");
         
    }
     
    return 0;
}