[bzoj3698]XWW的難題
阿新 • • 發佈:2019-01-05
inf pri urn 自己 min 小時 經典 git set
傳送門
有源匯上下界最大流,建模很經典,將每行和每列看做一個點,然後去按照有源匯上下界建圖就行了,因為這個題,我又發現了自己網絡流板子裏的一個錯誤寫法,bfs的時候得增廣完再退出,改了我半個小時。。。
#include<cstdio> #include<queue> #include<cstring> #include<iostream> #include<algorithm> using namespace std; #define rg register void read(int &x) { char ch; bool ok; for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1; for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x; } queue<int>q;int cnt=1,n,s,t,ans,inf=2e9,ss,sum,tt,dis[10001],in[10001],nxt[20001],pre[20001],h[10001],v[20001],cur[10001];double a[101][101]; inline void add(int x,int y,int z) { pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt,v[cnt]=z; pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt,v[cnt]=0; } inline bool bfs(int s,int t) { memset(dis,0,sizeof dis); q.push(s),dis[s]=1; while(!q.empty()) { int x=q.front();q.pop(); for(rg int i=h[x];i;i=nxt[i])if(v[i]&&!dis[pre[i]])dis[pre[i]]=dis[x]+1,q.push(pre[i]); } return dis[t]; } inline int dfs(int x,int flow,int t) { if(x==t||!flow)return flow; int f=flow; for(rg int &i=cur[x];i;i=nxt[i]) if(v[i]&&dis[pre[i]]==dis[x]+1) { int y=dfs(pre[i],min(v[i],f),t); f-=y,v[i]-=y,v[i^1]+=y; if(!f)return flow; } if(flow==f)dis[x]=-1; return flow-f; } int main() { scanf("%d",&n),s=0,t=2*n+1,ss=t+1,tt=ss+1; for(rg int i=1;i<=n;i++)for(rg int j=1;j<=n;j++)scanf("%lf",&a[i][j]); for(rg int i=1;i<n;i++) for(rg int j=1;j<n;j++) { if(a[i][j]!=(int)a[i][j])add(i,j+n,1); in[i]-=(int)a[i][j],in[j+n]+=(int)a[i][j]; } for(rg int i=1;i<n;i++) { if(a[i][n]!=(int)a[i][n])add(s,i,1); in[i]+=(int)a[i][n],in[s]-=(int)a[i][n]; } for(rg int i=1;i<n;i++) { if(a[n][i]!=(int)a[n][i])add(i+n,t,1); in[t]+=(int)a[n][i],in[i+n]-=(int)a[n][i]; } for(rg int i=0;i<=t;i++) { if(in[i]>0)add(ss,i,in[i]),sum+=in[i]; if(in[i]<0)add(i,tt,-in[i]); } add(t,s,inf); for(;bfs(ss,tt);ans+=dfs(ss,inf,tt))memcpy(cur,h,sizeof cur); if(sum!=ans){printf("No\n");return 0;} for(int i=h[ss];i;i=nxt[i])v[i]=v[i^1]=0; for(int i=h[tt];i;i=nxt[i])v[i]=v[i^1]=0; ans=0; for(;bfs(s,t);ans+=dfs(s,inf,t))memcpy(cur,h,sizeof cur); printf("%d\n",3*ans); }
[bzoj3698]XWW的難題