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值最大。
1<=n<=100,1<=a[i][j],b[i][j]<=10^4
来自 https://www.lydsy.com/JudgeOnline/problem.php?id=4819
Solution
求C的形式可以想到分数规划,二分一个答案mid,若mid比所求的最优解C要大则
∑ai−mid∑bi≤0
∑
a
i
−
m
i
d
∑
b
i
≤
0
,反之亦然
那么每次套mid进去做最大费用最大流,注意各种神奇的卡常优化以及精度问题
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))
const double EPS=0.00000001;
const int INF=100000000;
const int N=605;
const int E=640005;
int x[E],y[E],w[E],next[E];
int a[N][N],b[N][N];
int queue[N],pre[N];
int ls[N],edCnt;
bool vis[N];
double dis[N],c[E];
void add_edge(int tx,int ty,int tw,double tc) {
x[++edCnt]=tx; y[edCnt]=ty; w[edCnt]=tw; c[edCnt]=tc; next[edCnt]=ls[tx]; ls[tx]=edCnt;
x[++edCnt]=ty; y[edCnt]=tx; w[edCnt]=0; c[edCnt]=-tc; next[edCnt]=ls[ty]; ls[ty]=edCnt;
}
bool spfa(int st,int ed) {
int head=1,tail=1; queue[tail]=st;
rep(i,st,ed) dis[i]=-INF; dis[st]=0;
vis[st]=1;
while (head<=tail) {
int now=queue[head++];
for (int i=ls[now];i;i=next[i]) {
if (w[i]>0&&dis[now]+c[i]>dis[y[i]]) {
dis[y[i]]=dis[now]+c[i];
pre[y[i]]=i;
if (vis[y[i]]) continue;
vis[y[i]]=1;
queue[++tail]=y[i];
}
}
vis[now]=0;
}
return dis[ed]!=-INF;
}
void build_graph(int n,double C) {
rep(i,0,n*2+1) ls[i]=0; edCnt=1;
rep(i,1,n) {
rep(j,1,n) add_edge(i,j+n,1,-(double)b[i][j]*C+(double)a[i][j]);
add_edge(0,i,1,0);
add_edge(i+n,n*2+1,1,0);
}
}
double modify(int ed) {
int mn=INF;
for (int i=ed;pre[i];i=x[pre[i]]) {
mn=std:: min(mn,w[pre[i]]);
}
for (int i=ed;pre[i];i=x[pre[i]]) {
w[pre[i]]-=mn; w[pre[i]^1]+=mn;
}
return (double)mn*dis[ed];
}
double mcf(int n,double C,int st,int ed) {
build_graph(n,C);
double ret=0;
while (spfa(st,ed)) ret+=modify(ed);
return ret;
}
void solve(int n) {
double ans=-1;
for (double l=0,r=10000;r-l>=EPS;) {
double mid=(l+r)*0.5;
if (mcf(n,mid,0,n*2+1)<=0) r=mid;
else l=mid,ans=mid;
}
printf("%.6lf\n", ans);
}
int main(void) {
int n; scanf("%d",&n);
rep(i,1,n) rep(j,1,n) scanf("%d",&a[i][j]);
rep(i,1,n) rep(j,1,n) scanf("%d",&b[i][j]);
solve(n);
return 0;
}