1. 程式人生 > >有源匯帶上下界最小流

有源匯帶上下界最小流

  LOj 模板
思路我就不多說了吧,和有源匯帶上下界最大流一樣,只不過這次是初流-殘流網路最大流。關鍵這個模板題寫的過程無限T一組,讓我很絕望。在網上搜羅了很多程式碼,發現我有些地方可以優化。

  (1)跑dinic的時候可以使用當前弧優化

  (2)在dinic過程中,如果rest已經等於0了,直接返回。不要不管他,感覺沒什麼影響,其實有的資料會卡死的(千萬在邊權更新之後再返回)。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include
<vector> #include<queue> using namespace std; const int N=5e4+10; const int M=6e5+10; const int inf=1LL<<30; int ans,n,m,s,t,S,T,tot=1,sout,flow,maxflow,lin[N],in[M],A[N],d[N],cur[N]; struct node{int y,v,n;}e[M]; inline int read(){ char ch=getchar();int num=0,f=1; while(!isdigit(ch)){if
(ch=='-') f=-1;ch=getchar();} while(isdigit(ch)){num=(num<<1)+(num<<3)+(ch^48);ch=getchar();} return num*f; } inline void add(int x,int y,int w){ e[++tot].y=y;e[tot].n=lin[x];lin[x]=tot;e[tot].v=w; e[++tot].y=x;e[tot].n=lin[y];lin[y]=tot;e[tot].v=0; } inline bool bfs(int id){ memset(d,
0,sizeof(d)); queue<int>q; if(id==1) q.push(S),d[S]=1; else q.push(t),d[t]=1; while(q.size()){ int x=q.front();q.pop(); for(int i=lin[x];i;i=e[i].n){ int y=e[i].y; if(e[i].v&&!d[y]){ d[y]=d[x]+1; q.push(y); if(id==1&&y==T) return 1; if(id==2&&y==s) return 1; } } } return 0; } inline int dinic(int x,int flow,int id){ if(id==1&&x==T) return flow; if(id==2&&x==s) return flow; int rest=flow; for(int &i=cur[x];rest&&i;i=e[i].n){ int y=e[i].y; if(e[i].v&&d[y]==d[x]+1){ int k=dinic(y,min(rest,e[i].v),id); if(!k) d[y]=0; rest-=k; e[i].v-=k; e[i^1].v+=k; if(!rest) return flow-rest; } } return flow-rest; } int main(){ //freopen("h.in","r",stdin); //freopen("h.out","w",stdout); n=read(),m=read(),s=read(),t=read(); for(int i=1;i<=m;++i){ int x=read(),y=read(),l=read(),r=read(); add(x,y,r-l); A[y]+=l,A[x]-=l; //if(x==s) ans+=l; } S=0,T=n+1; for(int i=1;i<=n;++i){ if(A[i]>0) add(S,i,A[i]),sout+=A[i]; else if(A[i]<0) add(i,T,-A[i]); } add(t,s,inf); while(bfs(1)){ for(int i=0;i<=T;++i) cur[i]=lin[i]; while(flow=dinic(S,inf,1)) maxflow+=flow; } if(maxflow!=sout){ printf("please go home to sleep\n"); }else{ ans+=e[tot].v; maxflow=0; e[tot].v=e[tot^1].v=0; while(bfs(2)){ for(int i=0;i<=T;++i) cur[i]=lin[i]; while(flow=dinic(t,inf,2)) maxflow+=flow; } ans-=maxflow; printf("%d\n",ans); //prllf("%d\n",maxflow); } return 0; }