Codeforces Round #706 (Div. 2) B. Max and Mex

传送门

题目大意

给我们一个由n个不同的非负整数组成的multiset S，将元素⌈(a+b)/2⌉添加到S中，其中a=mex(S)， b=max(S)。如果这个数字已经在集合中，它将被再次添加。
multiset的max表示multiset中的最大整数，而multiset的mex表示multiset中不存在的最小非负整数t，例如
mex({1,4,0,2})=3;
mex({2,5,1})=0.
我们要找出计算k次操作之后S中不同元素的数量。

题解

如果mex(S)小于n，证明我们前n个数不连续，而我们加入的数并不会改变我们的mex和max，所以这种情况只用判断⌈(mex(S)+max(S))/2⌉在S中是否存在就好了，如果mex(S)大于等于n，证明前n个数全部连续，我们计算⌈(mex(S)+max(S))/2⌉会发现，每次的值都是最大值+1，所以运算几次，就会多几个不同的数

#include<cstdio>
const int MAX=1e5+5;
int a[MAX],have[MAX];
int main(){
	int t;
	int n,k,MEX,MAX;
	scanf("%d",&t);
	while(t--){
		MAX=0,MEX=0;
		scanf("%d%d",&n,&k);
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
			if(a[i]<n){
				have[a[i]]=1;
			}
			if(a[i]>MAX){
				MAX=a[i];
			}
		}
		while(have[MEX])++MEX;
		for(int i=0;i<n;i++){
			if(a[i]<n){
				have[a[i]]=0;
			}
		}
		int mid=(MEX+MAX+1)/2;
		if(mid>MAX){
			printf("%d\n",n+k);
		}else{
			int flag=1;
			for(int i=0;i<n;i++){
				if(mid==a[i]){
					flag=0;
				}
			}
			printf("%d\n",n+(flag&&k));
		}
	}
}