[BZOJ4819][SDOI2017]新生舞会(二分答案+费用流)

由于题目求的是最优配对方案,所以很容易想到建一个二分图后用费用流求解,即建立源点和汇点,由源点向每个男生连一条容量为 1 的边,由每个女生向汇点连一条容量为1的边,再从任意一个男生向任意一个女生连一条容量为 1 的边。
再考虑费用分配。这时候主要的问题在于C=ab这个式子中 b 在分母位置,无法简单求和。因此可以利用分数规划的思想去掉分母。先二分答案 ans
接下来判断最优解 C 能否大于或等于ans。把式子 C=ab 变形后为 aCb=0 。这时候就可以推出如果 Cans ,就一定存在一种 a,b 的合法分配方案,使得 aansb0 。此时就可以得出,源点连向男生,以及女生连向源点的边的费用都为 0 ,第i个男生向第 j 个女生的边的费用为ai,jansbi,j。建图完成后,如果最大费用最大流大于等于 0 ,那么Cans
代码:

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
inline int read() {
    int res = 0; bool bo = 0; char c;
    while (((c = getchar()) < '0' || c > '9') && c != '-');
    if (c == '-') bo = 1; else res = c - 48;
    while ((c = getchar()) >= '0' && c <= '9')
        res = (res << 3) + (res << 1) + (c - 48);
    return bo ? ~res + 1 : res;
}
const int L = 105, N = 5e5 + 5; const double eps = 1e-8;
int n, a[L][L], b[L][L], ecnt, nxt[N], adj[N], go[N], cap[N], len,
que[N], S, T; double cost[N], dis[N], ans;
bool vis[N], walk[N];
void add_edge(int u, int v, int w, double x) {
    nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v;
    cap[ecnt] = w; cost[ecnt] = x;
    nxt[++ecnt] = adj[v]; adj[v] = ecnt; go[ecnt] = u;
    cap[ecnt] = 0; cost[ecnt] = -x;
}
bool spfa() {
    int i; for (i = S; i <= T; i++) dis[i] = -1e20, walk[i] = 0;
    dis[que[len = 1] = S] = 0;
    for (i = 1; i <= len; i++) {
        int u = que[i]; vis[u] = 0;
        for (int e = adj[u], v; e; e = nxt[e])
            if (cap[e] > 0 && dis[u] + cost[e] > dis[v = go[e]]) {
                dis[v] = dis[u] + cost[e];
                if (!vis[v]) vis[que[++len] = v] = 1;
            }
    }
    return dis[T] >= -1e19;
}
int dfs(int u, int flow) {
    if (u == T) return ans += dis[T] * flow, flow;
    int res = 0, delta; walk[u] = 1;
    for (int e = adj[u], v; e; e = nxt[e])
        if (!walk[v = go[e]] && cap[e] > 0 &&
        abs(dis[v] - dis[u] - cost[e]) <= eps) {
            delta = dfs(v, min(cap[e], flow - res));
            if (delta) {
                cap[e] -= delta; cap[e ^ 1] += delta;
                res += delta; if (res == flow) break;
            }
        }
    return res;
}
double mcmf() {
    ans = 0;
    while (spfa()) dfs(S, 0x3f3f3f3f);
    return ans;
}
bool check(double mid) {
    int i, j; ecnt = S = 1; T = (n << 1) + 2;
    for (i = S; i <= T; i++) adj[i] = 0;
    for (i = 1; i <= n; i++) add_edge(S, i + 1, 1, 0);
    for (i = n + 2; i < T; i++) add_edge(i, T, 1, 0);
    for (i = 1; i <= n; i++) for (j = 1; j <= n; j++)
        add_edge(i + 1, j + n + 1, 1, 1.0 * a[i][j] - 1.0 * b[i][j] * mid);
    double ap = mcmf(); return abs(ap) <= eps || ap > eps;
}
int main() {
    int i, j; n = read();
    for (i = 1; i <= n; i++) for (j = 1; j <= n; j++)
        a[i][j] = read();
    for (i = 1; i <= n; i++) for (j = 1; j <= n; j++)
        b[i][j] = read();
    double l = -1e7, r = 1e7;
    while (abs(r - l) > eps) {
        double mid = (l + r) / 2;
        if (check(mid)) l = mid;
        else r = mid;
    }
    printf("%.6lf\n", r);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值