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