码迷,mamicode.com
首页 > 其他好文 > 详细

[SCOI2015] 小凸玩矩阵

时间:2019-08-16 22:51:19      阅读:77      评论:0      收藏:0      [点我收藏+]

标签:==   lan   简单的   thml   iostream   target   stream   简单   style   

题目链接

题意

  在一个 $n \times m$ 的矩阵 $A$ 中取出 $n$ 个数,任意两个数不能在同一行或同一列,求取出的第 $k$ 大数的最小值。

数据范围

  对于 $20\%$ 的数据,$1 \leq N \leq M \leq 9$

  对于 $40\%$ 的数据,$1 \leq N \leq M \leq 22$,$1 \leq N \leq 12$

  对于 $100\%$ 的数据,$1 \leq K \leq N \leq M \leq 250$,$1 \leq A_{i,j} \leq 10^9$

分析

  如果只是简单的取数,很容易想到把行和列分别放入两个点集做二分图

  但是这里要求了第 $k$ 大数最小,所以我们可以二分枚举答案,判断该值是否成立

  这也就相当于判断在小于等于该值的数中,能否取出合法的 $n-k+1$ 个数

  但如果是在小于该值的数中取出 $n-k$ 个数,这样判断是否可行

  实际上与枚举出的值相同的数不一定只有一个,这样做不能保证答案的最优性

技术图片
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 255
#define M (N * N << 1)

int n, m, k, ans, tot, maxflow;
int g[N][N], h[N * N], d[N << 1], cur[N << 1];
int to[M], cap[M], nxt[M], head[N << 1];

void add(int u, int v, int w) {
    to[++tot] = v; cap[tot] = w;
    nxt[tot] = head[u]; head[u] = tot;
    to[++tot] = u; cap[tot] = 0;
    nxt[tot] = head[v]; head[v] = tot;
}

bool bfs(int s, int t) {
    memset(d, 0, sizeof d);
    queue<int> q;
    d[s] = 1; q.push(s);
    while (!q.empty()) {
        int x = q.front(); q.pop();
        for (int i = head[x]; i; i = nxt[i])
            if (!d[to[i]] && cap[i]) {
                d[to[i]] = d[x] + 1;
                q.push(to[i]);
            }
    }
    return d[t];
}

int dfs(int x, int t, int res) {
    if (x == t || !res) return res;
    int flow = 0, f;
    for (int &i = cur[x]; i; i = nxt[i])
        if (d[to[i]] == d[x] + 1 && (f = dfs(to[i], t, min(cap[i], res)))) {
            flow += f; res -= f;
            cap[i] -= f; cap[i ^ 1] += f;
            if (!res) break;
        }
    return flow;
}

bool dinic(int now) {
    int s = 0, t = n + m + 1;
    tot = 1; maxflow = 0;
    memset(head, 0, sizeof head);
    for (int i = 1; i <= n; i++) add(s, i, 1);
    for (int i = 1; i <= m; i++) add(n + i, t, 1);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            if (g[i][j] <= now) add(i, n + j, 1);
    while (bfs(s, t)) {
        memcpy(cur, head, sizeof cur);
        maxflow += dfs(s, t, inf);
    }
    if (maxflow > n - k) return true;
    return false;
}

int main() {
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++) {
            scanf("%d", &g[i][j]);
            h[++h[0]] = g[i][j];
        }
    sort(h + 1, h + n * m + 1);
    int l = 1, r = n * m;
    while (l <= r) {
        int mid = (l + r) >> 1;
        if (dinic(h[mid])) ans = h[mid], r = mid - 1;
        else l = mid + 1;
    }
    printf("%d\n", ans);
    
    return 0;
}
View Code

[SCOI2015] 小凸玩矩阵

标签:==   lan   简单的   thml   iostream   target   stream   简单   style   

原文地址:https://www.cnblogs.com/Pedesis/p/11366439.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!