BZOJ 1305: [CQOI2009]dance跳舞 最大流

1305: [CQOI2009]dance跳舞

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

最大流

四排点。。。。突然不想写题解。。。

那就不写了。。。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return f*x;
}
const int N=300;const int inf=0X7f7f7f7f;
int n,k,ecnt=1,last[N],d[N],q[N],ans,a[60][60],S,fal,T=N-1; 
struct EDGE{int to,nt,val;}e[N*N];
inline void readd(int u,int v,int val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
inline void add(int u,int v,int val)
{readd(u,v,val);readd(v,u,0);}
bool bfs()
{
	memset(d,0,sizeof(d));
	int head=0,tail=1;q[0]=S;d[S]=1;
	while(head<tail)
	{
		int u=q[head++];
		for(int i=last[u];i;i=e[i].nt)
		if(e[i].val&&!d[e[i].to])
		{
			d[e[i].to]=d[u]+1;
			q[tail++]=e[i].to;
		}
	}
	return d[T];
}
int dfs(int u,int lim)
{
	if(u==T||!lim)return lim;
	int flow=0;
	for(int i=last[u];i;i=e[i].nt)
	if(d[e[i].to]==d[u]+1)
	{
		int tmp=dfs(e[i].to,min(e[i].val,lim));
		e[i].val-=tmp;e[i^1].val+=tmp;
		flow+=tmp;lim-=tmp;if(!lim)break;
	}
	if(!flow)d[u]=-1;
	return flow;
}
int dinic()
{while(bfs()){ans+=dfs(S,inf);}return ans;}
inline void build(int x)
{
	ecnt=1;memset(last,0,sizeof(last));ans=0;
	for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)if(a[i][j]==1)add(i,j+n,1);
	for(int i=1;i<=n;i++)add(i,i+(n<<1),k),add(S,i,x),add(i+n+(n<<1),i+n,k),add(i+n,T,x);
	for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)if(!a[i][j])add(i+(n<<1),j+n+(n<<1),1);
}
int main()
{
	n=read();k=read();char ch[60];int ans=n;
	for(int i=1;i<=n;i++){scanf("%s",ch+1);for(int j=1;j<=n;j++)if(ch[j]=='Y')a[i][j]=1;}
	int l=0,r=n,mid;
	while(l<=r){mid=l+r>>1;build(mid);if(dinic()==n*mid){fal=mid;l=mid+1;}else r=mid-1;}
	printf("%d\n",fal);
	return 0;
}