BZOJ 1715: [Usaco2006 Dec]Wormholes 虫洞 DFS版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"。

题解：裸的不能再裸的 SPFA 判负环. 用 DFS 据说能更快一点

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 10000
#define inf 10000000
using namespace std;
int hd[maxn],to[maxn<<1],nex[maxn<<1],val[maxn<<1];
int d[maxn],vis[maxn];
int edges,flag;
void add(int u,int v,int c)
{
    nex[++edges]=hd[u], hd[u]=edges, to[edges]=v, val[edges]=c;
}
void spfa(int u)
{
    int v;
    vis[u]=1;
    for(int i=hd[u];i;i=nex[i])
    {
        v=to[i];
        if(d[u]+val[i]<d[v])
        {
            if(vis[v] || flag)
            {
                flag=1;
                break;
            }
            d[v]=d[u]+val[i];
            spfa(v);
        }
    }
    vis[u]=0;
}
void re()
{
    memset(hd,0,sizeof(hd));
    memset(vis,0,sizeof(vis));
    for(int i=1;i<maxn;++i) d[i]=inf;
    edges=flag=0;
}
int main()
{
   //  setIO("input");
    int F;
    scanf("%d",&F);
    while(F--)
    {
        re();
        int n,m,w, a,b,c;
        scanf("%d%d%d",&n,&m,&w);
        for(int i=1;i<=m;++i)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b, c),add(b,a,c);
        }
        for(int i=1;i<=w;++i)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,-c);
        }
        for(int i=1;i<=n;++i)
        {
            d[i]=0;
            spfa(i);
            if(flag) break;
        }
        if(flag==1)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}