[bzoj3698]XWW的难题
传送门
有源汇上下界最大流,建模很经典,将每行和每列看做一个点,然后去按照有源汇上下界建图就行了,因为这个题,我又发现了自己网络流板子里的一个错误写法,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);
}