Codeforces 1684 E. MEX vs DIFF
題意
給你n個非負整數的數列a,你可以進行K次操作,每次操作可以將任意位置的數數更改成任意一個非負整數,求操作以後,DIFF(a)-MEX(a)的最小值;DIFF代表數組中數的種類。MEX代表數組中未出現的最小自然數。
提示
1. 顯然 DIFF(a)-MEX(a)最小,DIFF(a)越小越好,MEX(a)越大越好
2. 假如 DIFF 降低,同時 MEX 提升,這樣操作是不虧的,因此我們只需要提升MEX即可,貪心的的構造0-x,x為k次修改,能構建到mex的最大的數列a狀態。
3. 在原始a中,0-x中空缺的值即為需要填充個數的值,我們只需要貪心,先填入出現次數少的>x的值,以降低它的DIFF,即MEX固定了,再降低其DIFF
程式碼
#include<bits/stdc++.h>
using namespace std;
int a[100005], cnt[100005];
map<int, int> mp;
struct node {
int x, y;
} op[100005];
void run() {
int n, k;
cin >> n >> k;
for (int i = 0; i <= n; i++) {
cnt[i] = 0;
}
set<int> s;
mp.clear();
for (int i = 1; i <= n; i++) {
cin >> a[i];
if (a[i] < n)cnt[a[i]]++;
s.insert(a[i]);
mp[a[i]]++;
}
int res = 0, flag = n;
for (int i = 0; i < n; i++) {
if (cnt[i] == 0)res++;
if (res > k) {
flag = i;
break;
}
}
int st = 0;
for (auto [x, y]: mp) {
if (x >= flag)op[++st] = {x, y};
}
sort(op + 1, op + 1 + st, [&](const node &x, const node &y) { return x.y < y.y; });
int sum = 0;
int ree = min(res, k);
for (int i = 1; i <= st; i++) {
ree -= op[i].y;
if (ree >= 0)sum++;
else break;
}
cout << min(res, k) - sum + int(s.size()) - flag << '\n';
}
int main() {
int t;
cin >> t;
while (t--)
run();
return 0;
}