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);
}