Description
一次舞会有n个男孩和n个女孩。每首曲子开始时,所有男孩和女孩恰好配成n对跳交谊舞。每个男孩都不会和同一个女孩跳两首(或更多)舞曲。有一些男孩女孩相互喜欢,而其他相互不喜欢(不会“单向喜欢”)。每个男孩最多只愿意和k个不喜欢的女孩跳舞,而每个女孩也最多只愿意和k个不喜欢的男孩跳舞。给出每对男孩女孩是否相互喜欢的信息,舞会最多能有几首舞曲?
Input
第一行包含两个整数n和k。以下n行每行包含n个字符,其中第i行第j个字符为’Y’当且仅当男孩i和女孩j相互喜欢。
Output
仅一个数,即舞曲数目的最大值。
二分舞曲个数x,然后验证。
把每个人分成三个点,表示整体、喜欢的和不喜欢的。从原点向男孩整体连容量为x的边,从整体向喜欢连正无穷的边,向不喜欢连k的边。女孩也做同样处理。然后男女生互相喜欢的从喜欢节点向喜欢节点连1的边,互相不喜欢的从不喜欢节点到不喜欢节点连1的边,最后验证流量是否为x。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int oo=0x3f3f3f3f;
int fir[500],ne[10000],to[10000],w[10000],f[500],que[500],n,k,tot,s,t;
bool b[60][60];
/*boy: all:i like:i+n dis:i+2*n
girl: all:i+3*n like:i+4*n dis:i+5*n*/
void add(int num,int u,int v,int l)
{
ne[num]=fir[u];
fir[u]=num;
to[num]=v;
w[num]=l;
}
void init()
{
int i,j;
char ss[100];
scanf("%d%d",&n,&k);
s=6*n+1;
t=6*n+2;
for (i=1;i<=n;i++)
{
scanf("%s",ss+1);
for (j=1;j<=n;j++)
b[i][j]=ss[j]=='Y';
}
}
void build(int x)
{
int i,j;
for (i=1;i<=n;i++)
{
add(tot++,s,i,x);
add(tot++,i,s,0);
add(tot++,i,i+n,oo);
add(tot++,i+n,i,0);
add(tot++,i,i+n*2,k);
add(tot++,i+n*2,i,0);
add(tot++,i+3*n,t,x);
add(tot++,t,i+3*n,0);
add(tot++,i+n*4,i+n*3,oo);
add(tot++,i+n*3,i+n*4,0);
add(tot++,i+n*5,i+n*3,k);
add(tot++,i+n*3,i+n*5,0);
}
for (i=1;i<=n;i++)
for (j=1;j<=n;j++)
if (b[i][j])
{
add(tot++,i+n,j+n*4,1);
add(tot++,j+n*4,i+n,0);
}
else
{
add(tot++,i+n*2,j+n*5,1);
add(tot++,j+n*5,i+n*2,0);
}
}
bool find()
{
int hd=1,tl=1,u,v,i;
que[1]=s;
memset(f,0,sizeof(f));
f[s]=1;
while (hd<=tl)
{
u=que[hd++];
for (i=fir[u];i;i=ne[i])
if (w[i]&&!f[v=to[i]])
{
f[v]=f[u]+1;
que[++tl]=v;
}
}
return f[t];
}
int dfs(int u,int lim)
{
if (u==t) return lim;
int x,v,i,ret=0;
for (i=fir[u];i;i=ne[i])
if (w[i]&&f[v=to[i]]==f[u]+1)
{
x=dfs(v,min(lim-ret,w[i]));
w[i]-=x;
w[i^1]+=x;
ret+=x;
}
if (!ret) f[u]=0;
return ret;
}
bool ok(int x)
{
int i,j,ans=0,temp;
memset(fir,0,sizeof(fir));
memset(ne,0,sizeof(ne));
tot=2;
build(x);
while (find())
while (temp=dfs(s,oo))
ans+=temp;
return ans==x*n;
}
int main()
{
init();
int l=0,r=n,mid;
while (l<r)
{
mid=(l+r+1)/2;
if (ok(mid)) l=mid;
else r=mid-1;
}
printf("%d\n",l);
}