[bzoj3698]XWW的难题

Posted 2021-02-01 lcxer

传送门

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