BZOJ 1305: [CQOI2009]dance跳舞

题目地址：http://www.lydsy.com/JudgeOnline/problem.php?id=1305

题目大意：n个男孩和n个女孩，每个人和不喜欢的异性最多跳k次舞，问最多可以跳几次舞。

算法讨论：

        很容易看出这题是二分图。

        由于每个人可以和喜欢的异性跳任意次舞，那么一个人跳舞的最多次数为k+喜欢的异性数。

        于是源点向每个男孩连容量为k+喜欢的女孩数的边，每个女孩向汇点连容量为k+喜欢的男孩数的边，每个男生和每个女生之间连容量为1的边。

        然后最大流就是所求的答案。

        （据说还有更简单的贪心做法？）

Code：

/*
 * Problem:1305
 * Author:PYC
 */

#include <cstdio>
#include <queue>
#include <algorithm>

#define maxn 2000000
#define oo 1000000000

using namespace std;

char S[maxn];
int n,k,s,t,mm,ans,next[maxn+1],son[maxn+1],ed[maxn+1],cap[maxn+1],d[maxn+1],opp[maxn+1],boy[maxn+1],girl[maxn+1];

void add(int x,int y,int z){
	next[++mm]=son[x],son[x]=mm,ed[mm]=y,cap[mm]=z;
}

bool build(){
	queue<int> Q;
	for (int i=0;i<=2*n+1;++i) d[i]=-1;
	d[s]=0;
	Q.push(s);
	while (!Q.empty()){
		int x=Q.front();Q.pop();
		for (int i=son[x];i;i=next[i]){
			int y=ed[i];
			if (cap[i] && d[y]==-1){
				d[y]=d[x]+1;
				if (y==t) return 1;
				Q.push(y);
			}
		}
	}
	return 0;
}

int find(int x,int flow){
	if (x==t) return flow;
	int w=0,ret;
	for (int i=son[x];i && w<flow;i=next[i]){
		int y=ed[i];
		if (cap[i] && d[y]==d[x]+1 && (ret=find(y,min(flow-w,cap[i])))){
			cap[i]-=ret;
			cap[opp[i]]+=ret;
			w+=ret;
		}
	}
	if (!w) d[x]=-1;
	return w;
}

void dinic(){
	while (build()) while (find(s,oo));
}

int main(){
	scanf("%d%d",&n,&k);
	for (int i=1;i<=n;++i){
		scanf("%s",S);
		for (int j=1;j<=n;++j) if (S[j-1]=='Y') boy[i]++,girl[j]++;
	}
	for (int i=1;i<=n;++i){
		add(0,i,k+boy[i]),opp[mm]=mm+1;
		add(i,0,0),opp[mm]=mm-1;
		add(i+n,2*n+1,k+girl[i]),opp[mm]=mm+1;
		add(2*n+1,i+n,0),opp[mm]=mm-1;
	}
	for (int i=1;i<=n;++i)
		for (int j=1;j<=n;++j){
			add(i,j+n,1),opp[mm]=mm+1;
			add(j+n,i,0),opp[mm]=mm-1;
		}
	s=0,t=2*n+1;
	dinic();
	ans=oo;
	for (int i=son[s];i;i=next[i]) ans=min(ans,cap[opp[i]]);
	for (int i=son[t];i;i=next[i]) ans=min(ans,cap[i]);
	printf("%d\n",ans);
	return 0;
}

By Charlie Pan

Mar 15,2014