[BZOJ1305][CQOI2009]跳舞(二分答案+最大流)

首先,把每个男生和女生都拆成两个点,建立超级源汇 S,T
1、对于互相喜欢的一对男女生,由男生的第一个点向女生的第一个点连一条容量为 1 的边。
2、由所有男生的第一个点向第二个点连一条容量为k的边。
3、由所有女生的第二个点向第一个点连一条容量为 k 的边。
4、对于任意一对男女生,由男生的第二个点向女生的第二个点连一条容量为1的边。
然后二分最多舞曲数量 mid ,在二分过程中,
1、由 S 所有男生的第一个点连一条容量为mid的边。
2、由所有女生的第一个点向 T 连一条容量为mid的边。
跑一遍最大流,判断是否最大流 ==midn
代码:

#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 R = 55, N = 2e5 + 5, INF = 0x3f3f3f3f;
int n, K, S, T, ecnt, nxt[N], adj[N], go[N], cap[N], lev[N], len, que[N];
char s[R][R];
void add_edge(int u, int v, int w) {
    nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v; cap[ecnt] = w;
    nxt[++ecnt] = adj[v]; adj[v] = ecnt; go[ecnt] = u; cap[ecnt] = 0;
}
bool bfs() {
    int i; memset(lev, -1, sizeof(lev));
    lev[que[len = 1] = S] = 0;
    for (i = 1; i <= len; i++) {
        int u = que[i];
        for (int e = adj[u], v; e; e = nxt[e])
            if (cap[e] > 0 && lev[v = go[e]] == -1) {
                lev[que[++len] = v] = lev[u] + 1;
                if (v == T) return 1;
            }
    }
    return 0;
}
int dinic(int u, int flow) {
    if (u == T) return flow;
    int res = 0, delta;
    for (int e = adj[u], v; e; e = nxt[e])
        if (cap[e] > 0 && lev[u] < lev[v = go[e]]) {
            delta = dinic(v, min(cap[e], flow - res));
            if (delta) {
                cap[e] -= delta; cap[e ^ 1] += delta;
                res += delta; if (res == flow) break;
            }
        }
    if (res != flow) lev[u] = -1;
    return res;
}
int solve() {
    int res = 0;
    while (bfs()) res += dinic(S, INF);
    return res;
}
bool check(int mid) {
    int i, j; ecnt = 1;
    for (i = S; i <= T; i++) adj[i] = 0;
    for (i = 1; i <= n; i++) add_edge(S, i + 1, mid),
        add_edge(i + 3 * n + 1, T, mid), add_edge(i + 1, i + n + 1, K),
        add_edge(i + (n << 1) + 1, i + 3 * n + 1, K);
    for (i = 1; i <= n; i++) for (j = 1; j <= n; j++) {
        if (s[i][j] == 'Y') add_edge(i + 1, j + 3 * n + 1, 1);
        add_edge(i + n + 1, j + (n << 1) + 1, 1);
    }
    return solve() == mid * n;
}
int main() {
    int i, j, l = 1, r; r = n = read(); K = read();
    for (i = 1; i <= n; i++) scanf("%s", s[i] + 1);
    S = 1; T = (n << 2) + 2;
    while (l <= r) {
        int mid = l + r >> 1;
        if (check(mid)) l = mid + 1;
        else r = mid - 1;
    }
    printf("%d\n", r);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值