【每日一题】补档 CF1875 D. Jellyfish and Mex | 动态规划 | 中等

solego

已于 2024-04-15 13:53:34 修改

阅读量690

点赞数 30

分类专栏：算法竞赛文章标签：动态规划算法

于 2024-04-15 13:53:24 首次发布

本文链接：https://blog.csdn.net/weixin_43900869/article/details/137777658

版权

算法竞赛专栏收录该内容

61 篇文章 0 订阅

订阅专栏

文章讲解了如何通过动态规划解决一个涉及数组元素删除以减小mex值的问题，给出了解决方案并提供了C++代码示例。

摘要由CSDN通过智能技术生成

题目内容

原题链接

给定一个长度为 $n$ 的数组 $a$ ，每次选择一个元素 $a_i$ 删除，删除的代价为删除后剩余元素的 $m e x$ ， $m e x (a)$ 是指 $a$ 中未出现过的最小的非负数。

问将数组 $a$ 删除为空的操作的最小代价。

数据范围

$1\leq n\leq 5000$
$0\leq a_i\leq 10^9$

题解

考虑 $m e x (a)$ ，对于 $a$ 中大于 $m e x (a)$ 的数，必然放到最后使得 $m e x (a) = 0$ 时再操作，其操作代价为 $0$ 。

令数组 $a$ 的 $m e x (a)$ 为 $m$ ， $cnt_i$ 表示元素 $i$ 在 $a$ 中的出现次数。

那么我们只需要考虑所有小于等于 $m e x (a)$ 的数如何删除。

状态定义

$f_i$ 表示将 $m e x (a)$ 变成 $i$ 的最小操作代价。

状态转移

$f_i=\min\limits_{j=i+1}^m (f_j+j\times (cnt_i-1)+i)$

即我们考虑将 $m e x (a)$ 变成 $i$ 时，其之前的 $m e x (a)$ 必然大于 $i$ ，那么遍历 $j\in [i+1,m]$ 作为 $m e x (a)$ 。

对于 $m e x (a)$ 从 $j$ 降低到 $i$ 的情况，就是直接将所有的 $i$ 都删除。

对于前 $cnt_i-1$ 次对 $i$ 的删除操作，其代价为 $j$ ，对于第 $cnt_i$ 次对 $i$ 的删除操作，删除后的 $m e x (a)$ 已然变成了 $i$ ，其代价为 $i$ 。

故总代价为 $f_j+j\times (cnt_i-1)+i$ ，取所有 $j$ 的最小值即可。

最终答案为： $f_0$

时间复杂度： $O(n^2)$

代码

#include <bits/stdc++.h>
using namespace std;

void solve() {
    int n, x;
    cin >> n;

    vector<int> cnt(n);
    for (int i = 0; i < n; ++i) {
        cin >> x;
        if (x < n) cnt[x] += 1;
    }

    int mex = 0;
    while (mex < n && cnt[mex] > 0) mex += 1;

    // f[i] 表示使得 mex 变成 i 的最小操作次数
    const long long INF = 1e18;
    vector<long long> f(mex + 1, INF);
    f[mex] = 0;
    for (int i = mex - 1; i >= 0; --i)
        for (int j = i + 1; j <= mex; ++j)
            f[i] = min(f[i], f[j] + 1ll * (cnt[i] - 1) * j + i);

    cout << f[0] << "\n";
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T = 1;
    cin >> T;
    while (T--) {
        solve();
    }

    return 0;
}