[BZOJ]1305: [CQOI2009]dance跳舞 二分+网络流

Description

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

题解：

简单的二分+网络流，挺水的一道题，二分能有几首舞曲，然后用网络流判断，直接按照题意连边就好了，最多和k个不喜欢的跳舞这个条件，我是这样处理的，把每个男孩女孩拆成两个点，一个表示和它喜欢的跳舞，一个表示和它不喜欢的跳舞。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int inf=2147483647;
const int Maxn=60;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
struct Edge{int y,d,next;}e[500000];
int last[Maxn*4],len;
void ins(int x,int y,int d)
{
    int t=++len;
    e[t].y=y;e[t].d=d;
    e[t].next=last[x];last[x]=t;
}
void addedge(int x,int y,int d){ins(x,y,d);ins(y,x,0);}
int n,k;
bool in[Maxn*4];
queue<int>q;
int h[Maxn*4],st,ed;
bool bfs()
{
    memset(h,0,sizeof(h));h[st]=1;
    q.push(st);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=last[x];i;i=e[i].next)
        if(e[i].d&&!h[e[i].y])h[e[i].y]=h[x]+1,q.push(e[i].y);
    }
    return h[ed];
}
int dfs(int x,int f)
{
    if(x==ed)return f;
    int s=0,t;
    for(int i=last[x];i;i=e[i].next)
    {
        int y=e[i].y;
        if(h[y]==h[x]+1&&e[i].d&&s<f)
        {
            t=dfs(y,min(f-s,e[i].d));
            s+=t;e[i^1].d+=t;e[i].d-=t;
        }
    }
    if(s==0)h[x]=0;
    return s;
}
bool like[Maxn][Maxn];
//男生 1~n 男生2 n+1~2n  女生 2n+1~3n 女生2 3n+1~4n 
bool check(int x)
{
    if(x==0)return true;
    memset(last,0,sizeof(last));len=1;
    st=4*n+1;ed=4*n+2;
    for(int i=1;i<=n;i++)
    {
        addedge(st,i,x);
        addedge(i,i+n,k);
        for(int j=1;j<=n;j++)
        if(like[i][j])addedge(i,j+2*n,1);
        else addedge(i+n,j+3*n,1);
    }
    for(int i=1;i<=n;i++)
    {
        addedge(i+2*n,ed,x);
        addedge(i+3*n,i+2*n,k);
    }
    int ans=0;
    while(bfs())ans+=dfs(st,inf);
    return ans==n*x;
}
int main()
{
    n=read();k=read();
    for(int i=1;i<=n;i++)
    {
        char str[Maxn];
        scanf("%s",str+1);
        for(int j=1;j<=n;j++)
        if(str[j]=='Y')like[i][j]=true;
    }
    int l=0,r=n;
    while(l<=r)
    {
        int mid=l+r>>1;
        if(check(mid))l=mid+1;
        else r=mid-1;
    }
    printf("%d",l-1);
}