luogu 3385 dfs-spfa判断负环

(http://www.elijahqi.win/2017/07/10/luogu-3385-dfs-spfa%E5%88%A4%E6%96%AD%E8%B4%9F%E7%8E%AF/)
题目描述

暴力枚举/SPFA/Bellman-ford/奇怪的贪心/超神搜索

输入输出格式

输入格式：

第一行一个正整数T表示数据组数，对于每组数据：

第一行两个正整数N M，表示图有N个顶点，M条边

接下来M行，每行三个整数a b w，表示a->b有一条权值为w的边（若w<0则为单向，否则双向）

输出格式：

共T行。对于每组数据，存在负环则输出一行”YE5”(不含引号)，否则输出一行”N0”(不含引号)。

输入输出样例

输入样例#1：

2
3 4
1 2 2
1 3 4
2 3 1
3 1 -3
3 3
1 2 3
2 3 4
3 1 -8
输出样例#1：

N0
YE5
说明

N,M,|w|≤200 000；1≤a,b≤N；T≤10 建议复制输出格式中的字符串。

此题普通Bellman-Ford或BFS-SPFA会TLE

#include
#include
#define N 220000
int num,t,n,m,h[N],f[N];bool flag,visit[N];
struct node{
    int x,y,z,next;
}data[N<<1];
inline void insert1(int x,int y,int w){
    data[++num].x=x;data[num].y=y;data[num].z=w;data[num].next=h[x];h[x]=num;
    if (w>=0){
        data[++num].x=y;data[num].y=x;data[num].z=w;data[num].next=h[y];h[y]=num;
    }
}
void spfa(int x){
    visit[x]=true;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y,z=data[i].z;
        if (f[x]+zif (visit[y]||flag){
                flag=true;break;
            }
            spfa(y);
        }
    }
    visit[x]=false;
}
int main(){
    freopen("3385.in","r",stdin);
    //freopen("3385.out","w",stdout);
    scanf("%d",&t);
    while (t--){
        num=0;flag=false;memset(visit,false,sizeof(visit));memset(f,0,sizeof(f));
        scanf("%d%d",&n,&m);
        memset(h,0,sizeof(h));int x,y,w;
        for (int i=1;i<=m;++i) scanf("%d%d%d",&x,&y,&w),insert1(x,y,w);
        for (int i=1;i<=n;++i){
            spfa(i);if (flag) break;
        }
        if (flag) printf("YE5\n");else printf("N0\n");
    }
    return 0;
}