P3153 [CQOI2009]跳舞

题目描述

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

输入格式

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

输出格式

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

---

题解：
因为每个人都有两个状态，喜欢和不喜欢，所以我们选择将男女生的点都分成两个点，如果两个人是相互喜欢的，那么直接将男生的喜欢结点和女生的喜欢结点相连即可，边权为1，限制只能相互跳舞一次，不喜欢就将男生的不喜欢点连到女生的不喜欢点即可，一样边权为1，只能跳一次，那么中间的点就连完了

然后考虑建一个超级源点和超级汇点，超级源点连向每个男生的喜欢结点，超级汇点连向每个女生的喜欢结点，然后二分答案，设当前二分的值为mid，表示舞会最多能跳mid首舞曲，得到了mid，将超级源点和超级汇点连到的边的边权都赋为mid，表示每个男生最多跳mid次，然后建图跑dinic即可，因为每个人都要跳舞，不能空闲，所以如果答案是mid*n，表示相互之间男女生跳了mid * n次，这样才是符合条件的

---

AC代码：

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
#define LL long long
const int MAXN = 40000+50;
const int MAXM = 2e6+50;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int n,k,s,t,tot=1,head[MAXN],to[MAXM],nxt[MAXM],w[MAXM],h[MAXN];
char a[55][55];
inline void ade(int u,int v,int ww){
    to[++tot]=v; w[tot]=ww; nxt[tot]=head[u]; head[u]=tot;
}
inline void add(int u,int v,int w){ ade(u,v,w); ade(v,u,0); }
inline int bfs(){
    queue<int> que; que.push(s); memset(h,0,sizeof(h)); h[s]=1;
    while(!que.empty()){
        int u=que.front(); que.pop();
        for(int i=head[u];i;i=nxt[i]){
            if(w[i] && !h[to[i]]){
                h[to[i]]=h[u]+1; que.push(to[i]);
            }
        }
    }
    return h[t];
}
inline int dfs(int x,int f){
    if(x==t) return f; int fl=0;
    for(int i=head[x];i&&f;i=nxt[i]){
        if(w[i] && h[to[i]]==h[x]+1){
            int mi=dfs(to[i],min(f,w[i]));
            w[i]-=mi; w[i^1]+=mi; fl+=mi; f-=mi;
        }
    }
    if(!fl) h[x]=-1;
    return fl;
}
inline int dinic(){
    int res=0;
    while(bfs()) res+=dfs(s,INF);
    return res;
}
inline bool check(int mid){
    memset(head,0,sizeof(head)); tot=1;
    for(int i=1;i<=n;i++) add(s,i,mid),add(3*n+i,t,mid);
    for(int i=1;i<=n;i++) add(i,n+i,k),add(2*n+i,3*n+i,k);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(a[i][j]=='Y') add(i,3*n+j,1);
            else add(n+i,2*n+j,1);
    if(dinic()==mid*n) return true;
    return false;
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    scanf("%d%d",&n,&k); s=0,t=4*n+1;
    for(int i=1;i<=n;i++) scanf("%s",a[i]+1);
    int l=0,r=n+k,res=0;
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid)) l=mid+1,res=mid;
        else r=mid-1;
    }
    printf("%d\n",res);
    return 0;
}