【BZOJ4950】【WF2017】Mission Improbable（二分图匹配）

Description

click me

---

Solution

考虑满足的条件：
1. 对于俯视图：之前该方格内有箱子，那么最终至少要剩一个。
2. 对于每一行每一列的最高的那一个不能动。

我们考虑先将所有的箱子抽掉并保留每一行、每一列最高的格子，然后考虑哪些格子可以同时作为一行、一列的最高格子，这个显然可以用二分图匹配。注意：不同的最大值不会互相干扰，所以可以直接匹配。

---

Code

/************************************************
 * Au: Hany01
 * Prob: [BZOJ4950][WF2017] Mission Improbable
 * Date: May 16th, 2018
 * Email: hany01@foxmail.com
************************************************/

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;
#define File(a) freopen(a".in", "r", stdin), freopen(a".out", "w", stdout)
#define rep(i, j) for (register int i = 0, i##_end_ = (j); i < i##_end_; ++ i)
#define For(i, j, k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define x first
#define y second
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define ALL(a) (a).begin(), (a).end()
#define SZ(a) ((int)(a).size())
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define Mod (1000000007)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define y1 wozenmezhemecaia

template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read()
{
    register int _, __; register char c_;
    for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
    for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

const int maxn = 105;

LL Ans;
int n, m, lmax[maxn], cmax[maxn], a[maxn][maxn], w[maxn][maxn], vis[maxn], mat[maxn];

int dfs(int u)
{
    For(i, 1, m) if (w[u][i] && !vis[i]) {
        vis[i] = 1;
        if (!mat[i] || dfs(mat[i])) {
            mat[i] = u;
            return 1;
        }
    }
    return 0;
}

int main()
{
#ifdef hany01
    File("bzoj4950");
#endif

    n = read(), m = read();
    For(i, 1, n) For(j, 1, m) {
        chkmax(lmax[i], a[i][j] = read()), chkmax(cmax[j], a[i][j]);
        Ans += a[i][j] ? (a[i][j] - 1) : 0;
    }
    For(i, 1, n) Ans -= lmax[i] ? (lmax[i] - 1) : 0;
    For(i, 1, m) Ans -= cmax[i] ? (cmax[i] - 1) : 0;
    For(i, 1, n) For(j, 1, m)
        if (a[i][j] && lmax[i] == cmax[j]) w[i][j] = 1;

    For(i, 1, n) {
        Set(vis, 0);
        if (dfs(i)) Ans += lmax[i] - 1;
    }
    printf("%lld\n", Ans);

    return 0;
}
//明月别枝惊鹊，清风半夜鸣蝉。
//    -- 辛弃疾《西江月·夜行黄沙道中》