Codeforces 1364C - Ehab and Prefix MEXs

题意:给1e5的数组a 保证 ai <= ai+1  ai<=i

   求一个一样长的数组b 使得mex(b1,b2···bi) = ai 

QAQ:不知道为啥这1600分的题比赛时出不了 啊啊啊啊啊啊啊啊

题解:其实比赛的时候 就知道当前bi填的数字肯定是0,1,2...a[i] - 1中第一个没出现的数字

   如果已经被填满了 就在a[i] + 1...a[n]-1中填第一个后面没出现的数字

   因为显然有一个限制 你当前填的数字 不能等于后面的某一个a[i]

   比赛的时候一直想一些奇奇怪怪的实现方法.. 越整越麻烦

 

   显然bi要填满的数字就是0-a[n]-1

   把数字分成两组 一组是a[i]中出现的 一组是没出现的 分别维护指针

   如果a[i] > 第一组的数字 就填第二组的 ... 都填完了就选inf

   填完再判断一下合法性

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <string.h>
#include <queue>
using namespace std;
typedef long long ll;
const ll mod = 1e9 + 7;
const int MAXN = 1e5 + 5;

int val[100015];
int a[MAXN];
int b[MAXN];
int c[MAXN];
int d[MAXN];
int main() {
    int cn1 = 0, cn2 = 0;
    int n;
    cin>>n;
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]), val[a[i]]++;
    
    for(int i = 0; i < a[n]; i++) 
        if(!val[i]) d[++cn2] = i;
        else c[++cn1] = i;

    int nl = 1, nr = 1;
    for(int i = 1; i <= n; i++) {
        if(nl <= cn1 && a[i] > c[nl]) b[i] = c[nl++];
        else if(nr <= cn2) b[i] = d[nr++];
        else b[i] = n + 5; 
    }

    memset(val, 0, sizeof(val));
    int now = -1;
    for(int i = 1; i <= n; i++) {
        val[b[i]] = 1;
        while(val[now + 1]) now += 1;
        if(now + 1 != a[i]) {
            puts("-1");
            return 0;
        }
    }
    for(int i = 1; i <= n; i++) printf("%d ", b[i]); puts("");
    return 0;
}
View Code

 

   

posted @ 2020-06-14 11:39  lwqq3  阅读(577)  评论(0编辑  收藏  举报