2021牛客多校三 C_minimum grid

2021牛客多校三 C_minimum grid

题目传送门

题意：给出$n\times n$ 的格子，可以在其中m个给定位置的格子里填数，给出每行每列的最大值，要求填数和最小。

将行列限制放在一起，对于每个限制数，找到行列是这个限制数组成的子矩形，把每行每列建一个点，可以填数的位置建边，得到二分图，对这个二分图作最大匹配 cnt 就是能节省的个数，这个子矩阵对答案的贡献就是$(n^{\prime }+m^{\prime }-cnt)\times x$， 依次取出当前没有处理的数，建立二分图算贡献。

刚开始在想连边是不是还能是比他更大的限制的行列，需要对限制排序之类，后来发现不用，因为行列的限制是最大数是x而不是最大数不超过不小于x之类，这个x一定要填（

#include <bits/stdc++.h>

using namespace std;

constexpr int N = 2e3 + 10;
constexpr int M = 1e6 + 10;

vector<int> c[M], r[M], g[N];
int mp[N][N], mat[N];
bool vis[N];

inline bool dfs(int x) {
    for (int to: g[x]) {
        if (vis[to]) continue;
        vis[to] = 1;
        if (mat[to] == 0 || dfs(mat[to])) {
            mat[to] = x;
            return 1;
        }
    }
    return 0;
}

int main() {
    int n, m, k;
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 0, x; i < n; ++i) {
        scanf("%d", &x);
        r[x].push_back(i + 1);
    }
    for (int i = 0, x; i < n; ++i) {
        scanf("%d", &x);
        c[x].push_back(i + 1);
    }
    for (int i = 0, x, y; i < m; ++i) {
        scanf("%d%d", &x, &y);
        mp[x][y] = 1;
    }
    int ans = 0;
    for (int i = 1; i <= k; ++i) {
        for (int x: r[i]) {
            g[x].clear();
            for (int y : c[i]) {
                if (mp[x][y]) g[x].push_back(y);
            }
        }
        int cnt = 0;
        memset(mat, 0, sizeof mat);
        for (int x : r[i]) {
            memset(vis, 0, sizeof vis);
            if (dfs(x)) cnt++;
        }
        ans += (r[i].size() + c[i].size() - cnt) * i;
    }
    printf("%d", ans);
    return 0;
}