【bzoj 1305】[CQOI2009]dance跳舞

[CQOI2009]dance跳舞

Time Limit: 5 Sec Memory Limit: 162 MB
Submit: 3176 Solved: 1346
[Submit][Status][Discuss]
Description

一次舞会有n个男孩和n个女孩。每首曲子开始时，所有男孩和女孩恰好配成n对跳交谊舞。每个男孩都不会和同一个女孩跳两首（或更多）舞曲。有一些男孩女孩相互喜欢，而其他相互不喜欢（不会“单向喜欢”）。每个男孩最多只愿意和k个不喜欢的女孩跳舞，而每个女孩也最多只愿意和k个不喜欢的男孩跳舞。给出每对男孩女孩是否相互喜欢的信息，舞会最多能有几首舞曲？

Input

第一行包含两个整数n和k。以下n行每行包含n个字符，其中第i行第j个字符为’Y’当且仅当男孩i和女孩j相互喜欢。

Output

仅一个数，即舞曲数目的最大值。

Sample Input

3 0

YYY

YYY

YYY

Sample Output

3
HINT

N<=50 K<=30
二分网络流，每次检查是否满流，我好像建边建麻烦了

#include<cstdio>
#include<cstring>
#define inf 0x3f3f3f3f
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
char rec[100][100];
int n,k;
int h[1000*4],nx[50000],to[50000],cap[50000],cnt;
int lev[4000],q[4000],l,r;

void add_edge(int _u,int _v,int _c){cap[cnt]=_c;to[cnt]=_v;nx[cnt]=h[_u];h[_u]=cnt;cnt++;}
void link(int _u,int _v,int _c){/*printf("Link %d %d %d\n",_u,_v,_c);*/add_edge(_u,_v,_c);add_edge(_v,_u,0);}

bool bfs(int _s,int _t){
    l=r=0;memset(lev,-1,sizeof(lev));lev[_s]=0;q[r++]=_s;
    while(l<r){
        int t=q[l++];
        for(int i=h[t];i!=-1;i=nx[i]){
            if(lev[to[i]]==-1 && cap[i]>0){
                lev[to[i]]=lev[t]+1;
                q[r++]=to[i];
            }
        }
    }
//  printf("%d\n",lev[_t]);
    return lev[_t]!=-1;
}
int dfs(int _t,int _nd,int _fl){
//  printf("%d %d %d\n",_t,_nd,_fl);
    if(_t==_nd)return _fl;
    int w=0,used=0;
    for(int i=h[_nd];i!=-1;i=nx[i]){
        if(lev[to[i]]!=lev[_nd]+1 || cap[i]<=0)continue;
        w=_fl-used;w=dfs(_t,to[i],min(w,cap[i]));
        used+=w;cap[i]-=w;cap[i^1]+=w;
        if(used==_fl)return used;
    }
    return used;
}

int dinic(int _s,int _t){int rt=0;while(bfs(_s,_t))rt+=dfs(_t,_s,inf);return rt;}

void init(){
    memset(h,-1,sizeof(h));
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%s",rec[i]+1);

}
void rebuild(int _fl){
    memset(h,-1,sizeof(h));
    cnt=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(rec[i][j]=='Y'){
                link(3*(i-1)+1,3*n+3*(j-1)+1,1);
            }
            else{
                link(3*(i-1)+3,3*n+3*(j-1)+2,1);
            }
        }
    }
    for(int i=1;i<=n;i++){
        link(3*(i-1)+2,3*(i-1)+3,k);
    }
    for(int i=1;i<=n;i++){
        link(0,3*(i-1)+1,inf);
        link(0,3*(i-1)+2,inf);
    }
    for(int i=1;i<=n;i++){
        link(3*n+3*(i-1)+1,3*n+3*(i-1)+3,inf);
        link(3*n+3*(i-1)+2,3*n+3*(i-1)+3,inf);
    }
    for(int i=1;i<=n;i++){
        link(3*n+3*(i-1)+3,3*n+3*n+2,_fl);
    }

}
int solve(){
    int L=0,R=inf,ans=0;
    while(L<=R){
        int mid=(L+R)>>1;
//      printf("%d\n",mid);
        rebuild(mid);
        int fl=dinic(0,3*n+3*n+2);
        if(fl>=mid*n){
            ans=mid;
            L=mid+1;
        }
        else{
            R=mid-1;
        }
    }
    return ans;
}
int main(){
    freopen("pch.in","r",stdin);
    freopen("pch.out","w",stdout);
//  freopen("1.out","w",stdout);
    init();
    printf("%d",solve());
    return 0;
}