bzoj4819 [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值最大。

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要大则$\sum{a_i}-mid\sum{b_i}\leq 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;
}