SPFA判断图中是否有负环

使用SPFA算法判断图中首有负环。

SPFA算法保证每个顶点至多入队列n次（n为顶点总数）更新出到所有点的最短距离

如果某个点入队列次数大于n次表示图中有负环

#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;

const int N=1000;
const int M=10000;

struct edge
{
	int v,w,next;
	edge(){}
	edge(int _v,int _w,int _next)
	{
		v=_v;
		w=_w;
		next=_next;
	}
} e[M*2];

int head[N],size;

void initi()
{
	memset(head,-1,sizeof(head));
	size=0;
}

void insert1(int u,int v,int w)
{
	e[size]=edge(v,w,head[u]);
	head[u]=size++;
}

void insert2(int u,int v,int w)
{
	insert1(u,v,w);
	insert1(v,u,w);
}

int n,m;
int dis[N],cnt[N];//cnt记录每个顶点入队列的次数 
bool vis[N];// 


bool spfa(int u)
{
	memset(vis,false,sizeof(vis));
	memset(dis,0x3f,sizeof(dis));
	queue<int> q;
	q.push(u);
	vis[u]=true;
	dis[u]=0;
	while(!q.empty())
	{
		int t=q.front();
		q.pop();
		vis[t]=false;
		for(int i=head[t];i!=-1;i=e[i].next)
		{
			int v=e[i].v;
			int w=e[i].w;
			if(dis[t]+w<dis[v])
			{
				dis[v]=dis[t]+w; 
				if(!vis[v]) 
				{
					q.push(v);
					vis[v]=true;
					cnt[v]++;
					if(cnt[v]>n) return true;//如果某个顶点入队次数大于总顶点个数，则存在负环 
				}
			}
		}
	}
	return false;//不存在 
}

int main()
{
	initi();
	int u,v,w;
	scanf("%d%d",&n,&m);
	while(m--)
	{
		scanf("%d%d%d",&u,&v,&w);
		insert2(u,v,w);
	} 
	if(spfa(1)) printf("YES")
	else printf("NO");
	return 0;
}