Codeforces - 1139E - Maximize Mex（建图 + 二分图匹配）

1.1 题意

有 $n$ 名学生和 $m(1\le m\le n\le 5000)$ 个分组，每个学生 $i$ 都有能力 $p_i(0\le p_i<5000)$ 和其所在分组编号 $c_i(1\le c_i\le m)$。

比赛共有 $d(1\le d\le n)$ 天，第 $i$ 天有一名学生 $k_i$ 离开比赛，之后会在每一组中选择一名同学组成一个队伍参加比赛。当天队伍的能力为当天队伍中所有学生能力值的 $\text{mex}$。问每一天的队伍可以达到的最大能力。

一个多重集合 $S$ 的 $\text{mex}$ 指的是最小的不在 $S$ 中的非负整数。空集的 $\text{mex}$ 为 $0$。

1.2 解题过程

根据数据范围和题意，我们可以考虑以能力值为一侧，分组为另一侧建立一个二分图。

之后，对于还没有离开比赛的选手来说，将其能力值与其分组建边。

在某一天，如果我们按能力值从小到大对二分图进行匹配的话，会发现第一个匹配失败的能力值就是这一天的答案。

如果我们考虑时间倒流的话，会发现删人变成了加人，删边变成了加边，处理起来会方便很多。

更重要的是，按照时间倒序，答案一定是单调不减的。

因此我们设 $pos$ 为后一天的答案，则很显然处理当前天的时候，只需要从 $pos$ 开始匹配即可。

这样整个的时间复杂度就是完整地进行一次二分图匹配的时间复杂度，即 $O(nm)$，是可以接受的。

1.3 WA 点

如果二分图匹配中 $match[]$ 数组的初值为 $0$ 的话，则所有的能力值都要加 $1$ 之后建边，否则二分图匹配会发生错误。

1.4 代码

vector<int> G[maxn];
bool vis[maxn], mark[maxn];
int match[maxn];
int p[maxn], c[maxn], k[maxn];
int ans[maxn];
void addedge(int u, int v) {
    G[u].push_back(v);
}
bool dfs(int u) {
    for (auto v: G[u]) {
        if (!vis[v]) {
            vis[v] = true;
            if (!match[v] || dfs(match[v])) {
                match[v] = u;
                return true;
            }
        }
    }
    return false;
}
int work(int &id) {
    if (id >= 5000) return 5000;
    while (id < 5000) {
        for (int j = 0; j <= 10000; j++) {
            vis[j] = false;
        }
        if (!dfs(id)) return id;
        id++;
    }
    return 5000;
}
int main()
{
    int n, m, d;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) scanf("%d", &p[i]);
    for (int i = 1; i <= n; i++) scanf("%d", &c[i]);
    scanf("%d", &d);
    for (int i = 1; i <= d; i++) {
        scanf("%d", &k[i]);
        mark[k[i]] = true;
    }
    for (int i = 1; i <= n; i++) {
        if (!mark[i]) {
            addedge(p[i] + 1, c[i] + 5000);
            addedge(c[i] + 5000, p[i] + 1);
        }
    }
    int pos = 1;
    for (int i = d; i >= 1; i--) {
        ans[i] = work(pos);
        addedge(p[k[i]] + 1, c[k[i]] + 5000);
        addedge(c[k[i]] + 5000, p[k[i]] + 1);
    }
    for (int i = 1; i <= d; i++) printf("%d\n", ans[i] - 1);
    return 0;
}