bzoj1715 [Usaco2006 Dec]Wormholes 虫洞（spfa判负环）

spfa判负环，写的dfs版，听说更快些。注意d数组这回要清成0，可以更快。如果一个点在一条最短路径上出现了两次，则一定存在负环。

#include <bits/stdc++.h>
using namespace std;
#define N 510
#define ll long long
inline 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;
}
int tt,n,m,w,d[N],h[N],num;
bool vis[N],flag;
struct edge{
    int to,next,val;
}data[5300];
inline void add1(int x,int y,int v){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=v;
}
void spfa(int x){
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;
        if(d[x]+data[i].val<d[y]){
            if(vis[y]){flag=1;return;}//y在最短路径上出现了两次，必有负环 
            d[y]=d[x]+data[i].val;
            vis[y]=1;spfa(y);vis[y]=0;if(flag) return;
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    tt=read();
    while(tt--){
        memset(h,0,sizeof(h));num=0;
        n=read(),m=read(),w=read();
        while(m--){
            int x=read(),y=read(),v=read();add1(x,y,v);add1(y,x,v);
        }
        while(w--){
            int x=read(),y=read(),v=read();add1(x,y,-v);
        }
        memset(d,0,sizeof(d));memset(vis,0,sizeof(vis));flag=0;
        for(int i=1;i<=n;++i){vis[i]=1;spfa(i);vis[i]=0;if(flag) break;}
        if(flag) puts("YES");else puts("NO");
    }
    return 0;
}