P3385 【模板】判负环(DFS 跑SPFA)

既然我们只需要判断负环，那么就相当于我们需要找到一条权值和为负的回路。 
既然我们只需要找到权值和为负的回路，那不妨使距离数组d初始化为0。 
这样处理后，第一次拓展只会拓展到与起点相连边权为负的边。 
那么我们就分别枚举所有的点作为起点，如果已经找到一个负环就不再继续枚举。 
根据SPFA，我们找到的负环一定包含当前枚举的这个点。（因为这个点出现了两次啊） 
正确性相当的显然。
--

#include<cstdio>
#include<cstring>
#define INF 1e9+7
#define maxn 200005
#define maxm 200005
using namespace std;

struct data{int v,t,nxt;}edges[maxm<<1];
int head[maxn],cnt,d[maxn];
void insert(int a,int b,int v){edges[++cnt]=(data){b,v,head[a]};head[a]=cnt;}
bool vis[maxn],flag;

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

void DFS_SPFA(int u){
    if(flag) return ;
    vis[u]=true;
    for(int i=head[u];i;i=edges[i].nxt){
        if(flag) return ;
        int v=edges[i].v;
        if(d[u]+edges[i].t<d[v]){
            d[v]=d[u]+edges[i].t;
            if(vis[v]){
                flag=true;
                return ;
            }else{
                DFS_SPFA(v);
            }
        }
    }
    vis[u]=false;
}

int main(){
    int T=read();
    while(T--){
        int n=read(),m=read();
        memset(d,0,sizeof(d));
        memset(vis,0,sizeof(vis));
        memset(head,0,sizeof(head));
        cnt=0;
        flag=false;
        d[1]=0;

        for(int a,b,v,i=1;i<=m;i++){
            a=read();b=read();v=read();
            insert(a,b,v);
            if(v>=0){
                insert(b,a,v);
            }
        }
        for(int i=1;i<=n;i++){
            DFS_SPFA(i);
            if(flag) break;
        }
        if(flag){
            printf("YE5\n");
        }else{
            printf("N0\n");
        }
    }
    return 0;
}