1305: [CQOI2009]dance跳舞
Time Limit: 5 Sec Memory Limit: 162 MBSubmit: 2494 Solved: 1025
[ 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
YYY
YYY
YYY
Sample Output
3
HINT
N<=50 K<=30
Source
题解:二分+最大流
对舞曲数二分答案,然后判断是否满流(maxflow=mid(舞曲数)×n)
对于每个男孩和女孩分别拆成3个点,第一个点用来限制流量,第二个点表示与喜欢的女孩跳舞,第三个点表示与不喜欢的女孩跳舞,女孩也一样。具体方式如下图,蓝点表示不喜欢的点。如果是喜欢的点就把这个点到他的限流点的容量赋值为inf即可
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define N 10000
#define inf 1000000000
using namespace std;
int n,k,a[100][100];
int next[N],point[N],v[N],remain[N],tot;
int last[N],cur[N],num[N],deep[N],cnt,man;
char s[100];
void add(int x,int y,int z)
{
tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; remain[tot]=z;
tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; remain[tot]=0;
}
int addflow(int s,int t)
{
int now=t; int ans=inf;
while (now!=s)
{
ans=min(ans,remain[last[now]]);
now=v[last[now]^1];
}
now=t;
while (now!=s)
{
remain[last[now]]-=ans;
remain[last[now]^1]+=ans;
now=v[last[now]^1];
}
return ans;
}
void bfs(int s,int t)
{
for (int i=s;i<=t;i++)
deep[i]=t;
deep[t]=0;
queue<int> p; p.push(t);
while (!p.empty())
{
int now=p.front(); p.pop();
for (int i=point[now];i!=-1;i=next[i])
if (deep[v[i]]==t&&remain[i^1])
deep[v[i]]=deep[now]+1,p.push(v[i]);
}
}
int isap(int s,int t)
{
bfs(s,t);
for (int i=s;i<=t;i++)
cur[i]=point[i];
for (int i=s;i<=t;i++)
num[deep[i]]++;
int now=s; int ans=0;
while (deep[s]<t)
{
if (now==t)
{
ans+=addflow(s,t);
now=s;
}
bool f=false;
for (int i=cur[now];i!=-1;i=next[i])
if (deep[v[i]]+1==deep[now]&&remain[i])
{
last[v[i]]=i;
cur[now]=i;
f=true;
now=v[i];
break;
}
if (!f)
{
int minn=t;
for (int i=point[now];i!=-1;i=next[i])
if (remain[i]) minn=min(deep[v[i]],minn);
if (!--num[deep[now]]) break;
deep[now]=minn+1;
num[deep[now]]++;
cur[now]=point[now];
if (now!=s)
now=v[last[now]^1];
}
}
return ans;
}
void build(int x)
{
tot=-1;
memset(point,-1,sizeof(point));
memset(next,-1,sizeof(next));
for (int i=1;i<=n;i++)//1总数 ,2喜欢,3不喜欢
{
add(1,3*(i-1)+1+1,x);
add(3*(i-1)+1+1,3*i+1,k);
add(3*(i-1)+1+1,3*(i-1)+1+2,inf);
add(man+3*(i-1)+1,cnt,x);
add(man+3*i,man+3*(i-1)+1,k);
add(man+3*i-1,man+3*(i-1)+1,inf);
}
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
if (a[i][j])
add(3*(i-1)+1+2,man+3*(j-1)+2,1);
else
add(3*(i-1)+1+3,man+3*(j-1)+3,1);
}
int main()
{
scanf("%d%d\n",&n,&k);
memset(next,-1,sizeof(next));
memset(point,-1,sizeof(point));
tot=-1;
for (int i=1;i<=n;i++)
{
scanf("%s",s+1);
for (int j=1;j<=n;j++)
if (s[j]=='Y') a[i][j]=1;
}
man=3*n+1; cnt=6*n+2;
int head=0; int tail=n; int ans=0;
while (head<=tail)
{
int mid=(head+tail)/2;
build(mid);
if (isap(1,cnt)==n*mid)
ans=mid,head=mid+1;
else
tail=mid-1;
}
printf("%d\n",ans);
}