【模板】負環(spfa)
阿新 • • 發佈:2017-05-21
sizeof 貪心 com image 一行 clas 存在 cst -m
輸出樣例#1:
洛谷——P3385 【模板】負環
題目描述
暴力枚舉/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
N0 YE5
說明
N,M,|w|≤200 000;1≤a,b≤N;T≤10 建議復制輸出格式中的字符串。
此題普通Bellman-Ford或BFS-SPFA會TLE
bfs spfa判負環的代碼:
int spfa1(int s) { memset(dis,0x3f,sizeof(dis)); memset(vis,false,sizeof(vis)); queue<int>q; dis[s]=0,vis[s]=true; q.push(s); while(!q.empty()) {int x=q.front(); q.pop();vis[x]=false; for(int i=head[s];i;i=edge[i].next) { if(dis[x]+edge[i].ds<dis[edge[i].to]) { dis[edge[i].to]=dis[x]+edge[i].ds; sum[edge[i].to]++; q.push(edge[i].to);if(sum[edge[i].to]>n) return 1; } } } return 0; }
顯然,這樣是不對的,(不然我早就給全代碼了!)這樣會超時,全部TLE
這個題的正確是思路一個廣搜的spfa。
來,看看代碼(代碼中我會給出各步驟原因)
哎,終於很不容易的a了
#include<cstdio> #include<cstdlib> #include<cstring> #include<iostream> #include<algorithm> #define N 200003 using namespace std; int t,n,m,x,y,z,tot,dis[N],head[N]; bool vis[N],vist; struct Edge { int ds,to,next; }edge[N<<1]; void begin() { memset(dis,0,sizeof(dis)); memset(head,0,sizeof(head)); memset(vis,false,sizeof(vis)); tot=vist=0; } void add(int from,int to,int dis) { tot++; edge[tot].to=to; edge[tot].ds=dis; edge[tot].next=head[from]; head[from]=tot; } int read() { int x=0,f=1; char ch=getchar(); while(ch<‘0‘||ch>‘9‘) { if(ch==‘-‘) f=-1; ch=getchar(); } while(ch<=‘9‘&&ch>=‘0‘) { x=x*10+ch-‘0‘; ch=getchar(); } return x*f; } void spfa(int s) { vis[s]=true; for(int i=head[s];i;i=edge[i].next) { int x=edge[i].to; if(dis[s]+edge[i].ds<dis[x]) { dis[x]=dis[s]+edge[i].ds; if(vis[x]||vist) { vist=true; break; } spfa(x); } } vis[s]=false; } int main() { t=read(); while(t--) { n=read(),m=read(); begin(); for(int i=1;i<=m;i++) { x=read(),y=read(),z=read(); add(x,y,z); if(z>=0) add(y,x,z); } for(int i=1;i<=n;i++) { spfa(i); if(vist) break; } if(vist) printf("YE5\n"); else printf("N0\n"); } return 0; }
【模板】負環(spfa)