Acwing 852. spfa判斷負環
阿新 • • 發佈:2020-11-24
地址:https://www.acwing.com/problem/content/854/
解析:
引入cnt[i]陣列,表示到達當前這個點最短路的邊數。
對於一個正常的正權圖而言,每個點最多被更新n-1次,只會有n-1條邊。
如果到達某個點,最短路邊數>=n,那麼,說明有重複點,即,負權迴路。
由於SPFA原模板只針對了1號點。但是本題問的是是否存在,是有可能存在1號點到不了的負權迴路。
所以初始,把所有點加入佇列即可。
#include<iostream> #include<cstring> #include<algorithm> #include<queue> #include<stack> using namespace std; const int maxn=1e5+10; const int inf=0x3f3f3f3f; int h[maxn],e[maxn],ne[maxn],idx,w[maxn]; int dis[maxn],vis[maxn],cnt[maxn]; int n,m; void init() { memset(h,-1,sizeof(h)); memset(dis,inf,sizeof(dis)); dis[1]=0; } void add(int a,int b,int c) { w[idx]=c; e[idx]=b; ne[idx]=h[a]; h[a]=idx++; } int spfa() { queue<int>q; for(int i=1;i<=n;i++) { vis[i]=1; q.push(i); } while(!q.empty()) { int t=q.front(); q.pop(); vis[t]=0; for(int i=h[t];i!=-1;i=ne[i]) {int j=e[i]; if(dis[j]>dis[t]+w[i]) { dis[j]=dis[t]+w[i]; cnt[j]=cnt[t]+1; if(cnt[j]>=n) return true; if(!vis[j]) { q.push(j); vis[j]=1; } } } } return false; } int main() { init(); scanf("%d%d",&n,&m); for(int i=1;i<=m;i++) { int a,b,c; scanf("%d%d%d",&a,&b,&c); add(a,b,c); } if(spfa()) cout<<"Yes"<<endl; else cout<<"No"<<endl; }