SPFA查找负环 POJ3259

来源：互联网发布：淘宝网卡片手机编辑：程序博客网时间：2024/04/28 17:01

题目大意：给出n个顶点的无向图，并且在之后新增负边（虫洞）。问是否存在负环。

分析：使用spfa寻找负环。当一个节点入队列次数超过n-1时。表示存在负环。

#include <iostream>#include <cstring>#include <cstdio>#define MAXN 3000#define MAXM 3000using namespace std;struct Edge {int to, len, next;} edge[MAXM * 2];int en, head[MAXN];inline void add_edge(int u, int v, int len) {edge[en].to = v, edge[en].len = len, edge[en].next = head[u], head[u] = en++;}int n,m,k;int count[3000],q[1000000],f[3000];bool vis[3000];bool spfa(){int h,t,i;memset(vis,false,sizeof(vis));memset(count,0,sizeof(count));memset(f,10000,sizeof(f));h=1;t=1;q[1]=1;vis[1]=true;f[1]=0;count[1]=1;f[1]=0;while (h<=t){i=q[h];vis[i]=false;for (int p=head[i];p!=-1;p=edge[p].next){int to=edge[p].to,len=edge[p].len;if (f[to]>f[i]+len){f[to]=f[i]+len;if (!vis[to]){t++;q[t]=to;vis[to]=true;count[to]++;if (count[to]>=n) return true;}}}h++;}return false;}int main(){int cases,u,v,len;scanf("%d",&cases);while (cases--){scanf("%d%d%d",&n,&m,&k);fill(head+1,head+n+1,-1);en=0;for (int i=1;i<=m;i++){scanf("%d%d%d",&u,&v,&len);add_edge(u,v,len);add_edge(v,u,len);}for (int i=1;i<=k;i++){scanf("%d%d%d",&u,&v,&len);add_edge(u,v,-len);}if (spfa()) printf("YES\n");       else printf("NO\n");}return 0;}

0 0