［模板］负环---dfs版spfa

题目描述

暴力枚举/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

思路

　　dfs版spfa判环根据：若一个节点出现2次及以上，则存在负环．
　　｛补充bfs版本：若一个节点入队列的次数超过n，则存在负环．｝
　　如果是bfs来实现的话，果断到达上限（Ｔ-Ｌ-Ｅ），而dfs版的话，就相对快一些．
　　对于本题的强大数据来说，还需加入一定的优化：
　　由于是负环，所以无需像一般的spfa一样初始化为极大的数，只需要初始化为0就够了（可以减少大量的搜索，但要注意最开始时for一遍）

代码

#include <cstdio>#include <cstdlib>#include <cstring>#define open(s) freopen(s".in","r",stdin); freopen(s".out","w",stdout);#define close fclose(stdin); fclose(stdout); using namespace std;struct Edge{    int to;    int nxt;    int w;    inline void clr()    {        to=nxt=w=0;    }};int cnt;int head[200005];Edge edge[400005];inline void add(int x,int y,int z){    edge[++cnt]=(Edge){y,head[x],z};    head[x]=cnt;}int n,m;int dis[200005];bool vis[200005];inline int read(){    int k=1;    int sum=0;    char c=getchar();    for(;'0'>c || c>'9' ;c=getchar())        if(c=='-') k=-1;    for(;'0'<=c && c<='9';c=getchar())        sum=sum*10+c-'0';    return sum*k;}inline void write(int x){    if(x<0) { putchar('-'); x*=-1; }    if(x>9) write(x/10);    putchar(x%10+'0');}inline void clean()//初始化{    memset(dis,0,sizeof(dis));//dis初始化为0    memset(vis,0,sizeof(vis));    memset(head,0,sizeof(head));    for(;cnt;) edge[cnt--].clr();}inline bool dfs_spfa(int u)//dfs版的spfa(其实与bfs的差不多，只是改了搜索顺序){    vis[u]=1;    for(int i=head[u],v;i;i=edge[i].nxt)    {        v=edge[i].to;        if(dis[v]>dis[u]+edge[i].w)        {            dis[v]=dis[u]+edge[i].w;            if(vis[v] || dfs_spfa(v)) return 1;//如果访问过了（即找到负环）或者已经存在了负环        }    }    vis[u]=0;//注意改回来    return 0;//没找到}int main(){    open("3385");    for(int T=read();T;--T)    {        clean();        n=read(); m=read();        for(int i=1,x,y,z;i<=m;++i)        {            x=read(); y=read(); z=read();            add(x,y,z);            if(z>=0) add(y,x,z);        }        bool flag=0;        for(int i=1;i<=n && !flag;++i)            flag=dfs_spfa(i);        if(flag) printf("YE5"); else printf("N0");//神奇的输出        putchar('\n');    }    close;    return 0;}