当前位置: 代码迷 >> 综合 >> Codeforces C. Ehab and Prefix MEXs (思维) (Round #649 Div.2)
  详细解决方案

Codeforces C. Ehab and Prefix MEXs (思维) (Round #649 Div.2)

热度:19   发布时间:2023-12-22 13:46:51.0

传送门

题意: 给定长度为n的数组a,找到长度为n的另一个数组b,使得:
对于每个i(1≤i≤n)MEX({b1,b2,…,bi})= ai。
一组整数的MEX是不属于该组的最小非负整数。
如果这样的数组不存在,输出-1。
在这里插入图片描述
思路:

  • 为了满足a[i]是b1 ~ bi未出现的最小非负整数,则b[i]得为ai ~ n未出现的尽量小的数。
  • 这便可以用set统计整个a数组中未出现的数,从1到n开始枚举,每次取set的第一个元素,并将其删除;若i >1时且a[i] != a[i - 1],还可以把a[i - 1]插入set内,以达到去尽量小的目的。

代码实现:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {
    {
    1, 0}, {
    -1, 0}, {
    0, 1}, {
    0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;int n, m;
int a[N], b[N];
map<int, int> vis;
set<int> tmp;signed main()
{
    IOS;cin >> n;for(int i = 0; i < n; i ++){
    cin >> a[i];vis[a[i]] ++;}int m = n * 2;for(int i = 0; i < m; i ++)if(!vis[i]) tmp.insert(i);for(int i = 0; i < n; i ++){
    if(i && a[i] != a[i - 1]) tmp.insert(a[i - 1]);b[i] = *tmp.begin();tmp.erase(*tmp.begin());}for(int i = 0; i < n; i ++)cout << b[i] << " ";return 0;
}
  相关解决方案