C. Meximum Array(贪心 + 数学mex)

题目大意 : 给定一个长度为n的数组, 每次可以取前k个数求一个mex放入b并删除这k个数, 要求使b的字典序最大的操作方法, 输出b数组

思路 : 一道经典的运用mex的性质的题目,可以用一个数组来预处理以下后缀mex(因为每次操作的是前缀mex, 并不会对后缀mex产生影响), 由mex的性质可知, 第 i 位的后缀mex存的即为当前序列能取到的最大的mex, 要使b最大, 则取mex时应取得当前序列得最大mex, 即刚刚求得后缀mex, 所以做法就是跑一遍前缀mex, 当前缀mex取到当前得最大值时进行更新

(后缀mex : 倒着看,从 n 到 i 得序列得mex; 前缀mex : 就是题目描述的mex)

这里采用了一个聚聚自己写的mex的板子(tql), 具体可以去看看他的文章:

序列中未出现的最小的非负整数 mex 入门

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <iterator>
#include <unordered_set>
#include <cmath>
#include <algorithm>
#include <numeric>
#include <sstream>
#include <map>
#include <stack>
#include <set>
#include <queue>
#include <iomanip>
#include <bitset>
#include <unordered_map>
using namespace std;

stringstream ss;
#define endl "\n"
typedef long long ll;
typedef pair<int, int> PII;
const int N = 5e5+10, mod = 1e9+10;
const int INF = 0x3f3f3f3f;

// MEX的数据结构板子, 这里没有用到删除操作, 所以不用开set来维护
struct MEX
{
    int cnt[N];
    vector<int> res;
    int id = 0;
    void add(int x)
    {
        res.push_back(x);
        cnt[x]++;
    }
    int mex()
    {
        while(cnt[id]) id++;
        return id;
    }
    void clear()
    {
        id = 0;
        for(int x : res) cnt[x]--;
        res.clear();
    }
};

MEX pre, t;
// last来存后缀mex
int n, last[N], a[N];

void solve()
{
    t.clear(), pre.clear();
    cin>>n;
    for(int i = 1; i<=n; i++) cin>>a[i];
    for(int i = n; i>=1; i--)
    {
        t.add(a[i]);
        last[i] = t.mex();
    }
    int mx = last[1];
    vector<int> ans;
    for(int i = 1; i<=n; i++)
    {
        pre.add(a[i]);
        if(pre.mex() == mx)
        {
            ans.push_back(mx);
            mx = last[i+1];
            pre.clear();
        }
    }
    cout<<ans.size()<<endl;
    for(auto t : ans) cout<<t<<" ";
    cout<<endl;
}

int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
//    solve();
    int T;
    cin>>T;
    while(T -- )
    {
        solve();
    }
}