［模板］负环---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;
}