[BZOJ 4819] [Sdoi2017]新生舞会 二分+网络流

Description

学校组织了一次新生舞会，Cathy作为经验丰富的老学姐，负责为同学们安排舞伴。有n个男生和n个女生参加舞会

买一个男生和一个女生一起跳舞，互为舞伴。Cathy收集了这些同学之间的关系，比如两个人之前认识没计算得出 

a[i][j] ，表示第i个男生和第j个女生一起跳舞时他们的喜悦程度。Cathy还需要考虑两个人一起跳舞是否方便，

比如身高体重差别会不会太大，计算得出 b[i][j]，表示第i个男生和第j个女生一起跳舞时的不协调程度。当然，

还需要考虑很多其他问题。Cathy想先用一个程序通过a[i][j]和b[i][j]求出一种方案，再手动对方案进行微调。C

athy找到你，希望你帮她写那个程序。一个方案中有n对舞伴，假设没对舞伴的喜悦程度分别是a'1,a'2,...,a'n，

假设每对舞伴的不协调程度分别是b'1,b'2,...,b'n。令

C=(a'1+a'2+...+a'n)/(b'1+b'2+...+b'n),Cathy希望C值最大。

Input

第一行一个整数n。

接下来n行，每行n个整数，第i行第j个数表示a[i][j]。

接下来n行，每行n个整数，第i行第j个数表示b[i][j]。

1<=n<=100,1<=a[i][j],b[i][j]<=10^4

Output

一行一个数，表示C的最大值。四舍五入保留6位小数，选手输出的小数需要与标准输出相等

Sample Input

3
19 17 16
25 24 23
35 36 31
9 5 6
3 4 2
7 8 9

Sample Output

5.357143

思路

C=Σa/Σb 即 Σa=Σb*C  所以有 (a1-C*b1)+(a2-C*b2)+(a2-C*b2)+……+(an-C*bn)=0

可以想到二分 C 判断当前C是否可行 又因为每一个i都对应一个j 不能重复选 那么很自然的想到用网络流

因为C越大(a1-C*b1)+(a2-C*b2)+(a2-C*b2)+……+(an-C*bn)越小 那么如果当前的最大值(a1-C*b1)+(a2-C*b2)+(a2-C*b2)+……+(an-C*bn)要大于0的话 就说明C可以变大一点

建图

i∈男生 j∈女生

设立 源点s 汇点 t 从s到 i连边 (1,0) 从 每一个i向每一个(1,a[i][j]-C*b[i][j])的边 从每一个 j向 t连一条(1,0)的边 跑一遍最小费用最

大流

注意精度！

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn=500+5;
int a[maxn][maxn],cur[maxn][maxn],b[maxn][maxn];
int n;
int s,t;
bool visit[maxn];
double l,r,fee[maxn][maxn],ans;
int pre[maxn],q[maxn*maxn];
double dis[maxn];
bool can(double x);
bool spfa();
void EK();
int main()
{
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
	l=0,r=10000;
	scanf("%d",&n);
	s=2*n+1; t=s+1;
	for(int i=1; i<=n; i++)
	{
		for(int j=1; j<=n; j++)
		{
			scanf("%d",&a[i][j]);
		}
	}
	for(int i=1; i<=n; i++)
	{
		for(int j=1; j<=n; j++)
		{
			scanf("%d",&b[i][j]);
		}
	}
	while(!(fabs(l-r)<1e-10))
	{
		double mid=(l+r)/2;
		if(can(mid)) l=mid;
		else r=mid-(1e-7);
	}
	printf("%.6lf",l+1e-7);
	return 0;
}
bool can(double x)
{	
	memset(cur,0,sizeof(cur));
	for(int i=1; i<=n; i++)
	{	
		cur[s][i]=1;
		for(int j=1; j<=n; j++)
		{
			cur[i][j+n]=1;
			fee[i][j+n]=a[i][j]-double(x*b[i][j]);
			fee[j+n][i]=-fee[i][j+n];
		}
		cur[i+n][t]=1;
	
	}
/*	for(int i=1; i<=n*2; i++)
		{
			for(int j=1; j<=n*2; j++)
			{
				printf("%d ",cur[i][j]);
			}
			puts("");
		}*/
	ans=0;
	EK();
/*	cout<<ans<<' '<<l<<' '<<r<<endl;
		puts("@@@@@@@@@@@@");*/
	if(ans<0) return false;
	return true;
}
void EK()
{
	while(spfa())
	{
		int flow=0x7ffffff;
		int p=t;
		while(pre[p]!=-1)
		{
			cur[pre[p]][p]-=1;
			cur[p][pre[p]]+=1;
			p=pre[p];
		}
		ans+=dis[t];
	}
}
bool spfa()
{	
	for(int i=1; i<=t; i++)
	{
		dis[i]=-1e20;
	}
	memset(pre,-1,sizeof(pre));
	int head=0,tail=1;
	q[tail]=s; dis[s]=0;
	bool flag=false;
	while(head<tail)
	{
		head++; int now=q[head]; visit[now]=false;
		if(now==t) flag=true;
		for(int i=1; i<=t; i++)
		{
			if(cur[now][i] && dis[i]<dis[now]+fee[now][i])
			{
				dis[i]=dis[now]+fee[now][i]; pre[i]=now;
//				cout<<dis[now]+fee[now][i]<<endl;
				if(!visit[i])
				{
					visit[i]=true; tail++; q[tail]=i;
				}
			}
		}
	}
	return flag;
}