bzoj千题计划130：bzoj1305: [CQOI2009]dance跳舞

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>

using namespace std;

#define N 2501
#define M 3001

int n,k;
char s[51];
bool mp[51][51];

int tot;
int front[N],to[M<<1],nxt[M<<1],val[M<<1];

int src,decc;

int cur[N],lev[N];

queue<int>q;

void read(int &x)
{
    x=0;  char c=getchar();
    while(!isdigit(c)) c=getchar(); 
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar();  }
}

void add(int u,int v,int w)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; val[tot]=w;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; val[tot]=0;
}

void rebuild(int mid)
{
    tot=1;
    memset(front,0,sizeof(front));
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
            if(mp[i][j]) add(i,n*3+j,1);
            else add(n+i,n*2+j,1);
    for(int i=1;i<=n;++i) add(src,i,mid);
    for(int i=1;i<=n;++i) add(n*3+i,decc,mid);
    for(int i=1;i<=n;++i) add(i,n+i,k);
    for(int i=1;i<=n;++i) add(n*2+i,n*3+i,k);
}

bool bfs()
{
    while(!q.empty()) q.pop();
    for(int i=src;i<=decc;i++) lev[i]=-1,cur[i]=front[i];
    lev[src]=0; 
    q.push(src);
    int now;
    while(!q.empty())
    {
        now=q.front(); q.pop();
        for(int i=front[now];i;i=nxt[i])
        {
            if(lev[to[i]]==-1&&val[i]>0)
            {
                lev[to[i]]=lev[now]+1;
                if(to[i]==decc) return true;
                q.push(to[i]);
            }
        }
    }
    return false;
}
int dinic(int now,int flow)
{
    if(now==decc) return flow;
    int rest=0,delta;
    for(int &i=cur[now];i;i=nxt[i])
    {
        if(lev[to[i]]>lev[now]&&val[i]>0)
        {
            delta=dinic(to[i],min(flow-rest,val[i]));
            if(delta)
            {
                val[i]-=delta; val[i^1]+=delta;
                rest+=delta; if(rest==flow) break;
            }
        }
    }
    if(rest==flow) lev[now]=-1;
    return rest;
}

bool check(int mid)
{
    rebuild(mid);
    int maxflow=0;
    while(bfs()) 
    maxflow+=dinic(src,2e9);
    return maxflow==n*mid;
}

int main()
{
    read(n); read(k);
    decc=n*4+1;
    for(int i=1;i<=n;++i)
    {
        scanf("%s",s+1);
        for(int j=1;j<=n;++j) mp[i][j]=s[j]==‘Y‘ ? true : false; 
    }
    int l=0,r=n,mid,ans;
    while(l<=r)
    {
        mid=l+r>>1;
        if(check(mid)) ans=mid,l=mid+1;
        else r=mid-1; 
    }
    cout<<ans;
}