Codeforces Round #762 (Div. 3 E)

传送门

题目描述

输入描述

输出描述

输入样例

5
3
0 1 3
7
0 1 2 3 4 3 2
4
3 0 0 0
7
4 6 2 3 5 0 5
5
4 0 1 0 4

输出样例

1 1 0 -1 
1 1 2 2 1 0 2 6 
3 0 1 4 3 
1 0 -1 -1 -1 -1 -1 -1 
2 1 0 2 -1 -1 

样例解释

---

题目大意： 给定一个数组，可对其中的任一元素做无限次 +1 操作，当该数组满足其 mex 函数的结果分别为 $0n$ 时，求需要的最少操作次数。

对比 $mex=x$ 和 $mex=x+1$，可以发现在 $mex=x$ 中，$[0,x-1]$ 项都已经排好，则 $mex=x+1$ 可以沿用该规律，将第 x 项排好即可，并将所有的 $x+1$ 都变为 $x+2$ 。

因此可使用数组记载对应需要的操作次数，例如 $cnt[x]$ 记录排好前 $[0,x-1]$ 项所需要的操作次数，最后的答案为 $cnt[x]+mp[x]$ 。而 $cnt[x+1]$ 将多记录同时排好 x 需要的操作次数，若原数组存在 x ，则 $cnt[x+1]=cnt[x]$ ，否则 x 需要从之前多余的数进行若干次 +1 操作补上。

若要满足 $mex=x$ 存在，则小于 x 的元素数必须要大于等于 x，同时将多次出现的元素存入 vector 中，需要填补时，从 vector 中取最大的元素，并在操作次数中加上差值即可。

参考代码

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

const int N=2e5+10;
ll a[N],s[N],cnt[N];
map<ll,ll> mp;
vector<ll> v;

void init(ll n){
	for(ll i=0;i<=n+1;i++){
		mp[i]=0;
		a[i]=0;
		s[i]=0;
		cnt[i]=0;
	}
}

int main(){
	ll t;
	cin>>t;
	while(t--){
		ll n;
		cin>>n;
		init(n);
		for(ll i=1;i<=n;i++){
			cin>>a[i];
			mp[a[i]]++;
		}
		v.clear();
		sort(a+1,a+n+1);
		s[0]=0;
		for(ll i=1;i<=n;i++)//数组s：小于当前数的个数 
			s[i]=s[i-1]+mp[i-1];


		cout<<mp[0]<<" ";
		if(mp[0]>1)
			for(ll i=1;i<mp[0];i++)
				v.push_back(0);
			
		for(ll i=1;i<=n;i++){			
			if(s[i]<i){//之后全为-1 
				for(ll j=i;j<=n;j++)
					cout<<"-1 ";
				break;
			}
			else{
				ll ans=cnt[i-1];
				if(mp[i-1]==0){//补上缺失的i-1 
					//cout<<v.size()<<endl;
					ans+=((i-1)-v.back());
					v.pop_back();
				}
				cout<<ans+mp[i]<<" ";
				cnt[i]=ans;
			}
			if(mp[i]>1){
				for(ll j=1;j<mp[i];j++){
					v.push_back(i);//将多出的i放入可改变队列v 
				}
			}
		}
		cout<<endl;
	}
	return 0;
}