[CQOI2009][BZOJ1305] dance跳舞

1305: [CQOI2009]dance跳舞

Time Limit: 5 Sec  Memory Limit: 162 MB
Submit: 1877  Solved: 799
[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

Source

加强数据By dwellings and liyizhen2

又是二分网络流……这几天真敲网络流敲恶心了。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define MAXN 300
#define MAXE 100000
#define INF 100000007
using namespace std;
char c[100];
int n,k,mx,T,tot,cnt,head[MAXN],dis[MAXN],q[MAXN];
int next[MAXE],key0[MAXE],key[MAXE],list[MAXE];
void insert(int x,int y,int z)
{
    next[++tot]=head[x];
    head[x]=tot;
    list[tot]=y;
    key0[tot]=z;
}
void rebuild(int x)
{
    for (int i=2;i<=cnt;i++)
        key[i]=key0[i];
    for (int i=cnt+1;i<=tot;i+=2)
        key[i]=x;
    for (int i=cnt+2;i<=tot;i+=2)
        key[i]=0;
}
bool BFS()
{
    memset(dis,0xff,sizeof(dis));
    int t=0,w=1,x;
    dis[0]=1;q[1]=0;
    while (t<w)
    {
        x=q[++t];
        for (int i=head[x];i;i=next[i])
            if (key[i]&&dis[list[i]]==-1)
            {
                dis[list[i]]=dis[x]+1;
                q[++w]=list[i];
            }
    }
    return dis[T]!=-1;
}
int find(int x,int flow)
{
    if (x==T) return flow;
    int used=0,w;
    for (int i=head[x];i;i=next[i])
        if (key[i]&&dis[list[i]]==dis[x]+1)
        {
                                            w=find(list[i],min(key[i],flow-used));
                                            key[i]-=w;
                                            key[i^1]+=w;
                                            used+=w;
                                            if (used==flow) return flow;
        }
    if (!used) dis[x]=-1;
    return used;
}
bool dinic(int x)
{
    int ans=0;
    while (BFS()) ans+=find(0,INF);
    if (ans>=x*n) return 1; return 0;
}
int main()
{
    scanf("%d%d",&n,&k);
    T=4*n+1;
    tot=1;
    for (int i=1;i<=n;i++)
    {
        scanf("%s",c);
        for (int j=0;j<n;j++)
            if (c[j]==‘Y‘)
            {
                insert(i,j+1+3*n,1);
                insert(j+1+3*n,i,0);
            }
            else
            {
                insert(i+n,j+1+2*n,1);
                insert(j+1+2*n,i+n,0);
            }
    }
    for (int i=1;i<=n;i++)
    {
        insert(i,i+n,k);
        insert(i+n,i,0);
        insert(i+2*n,i+3*n,k);
        insert(i+3*n,i+2*n,0);
    }
    cnt=tot;
    for (int i=1;i<=n;i++)
    {
        insert(0,i,0);
        insert(i,0,0);
        insert(i+3*n,T,0);
        insert(T,i+3*n,0);
    }
    int l=0,r=n;
    while (l<=r)
    {
        int mid=(l+r)>>1;
        rebuild(mid);
        if (dinic(mid))
        {
            mx=mid;
            l=mid+1;
        }
        else r=mid-1;
    }
    printf("%d",mx);
    return 0;
}