bzoj 1305: [CQOI2009]dance跳舞（二分+最大流）

1305: [CQOI2009]dance跳舞

Time Limit: 5 Sec   Memory Limit: 162 MB
Submit: 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

Sample Output

3

HINT

N<=50 K<=30

Source

加强数据By dwellings and liyizhen2

[ Submit][ Status][ Discuss]

题解：二分+最大流

对舞曲数二分答案，然后判断是否满流（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);
}