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

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

题目

题目大意

刚开始给你n个不同的数，k次询问每次求在该序列中已存在的最大的数和不存在的最小的数（大于等于零）的和除以2想上取整在插入原序列中，经过k次询问后求序列中有多少个不同的元素。

输入样例

5
4 1
0 1 3 4
3 1
0 1 4
3 0
0 1 4
3 2
0 1 2
3 2
1 2 3

输出样例

4
4
3
5
3

解题思路

题目给你k的范围最大到1e9，那么肯定就是有规律可寻的，一共有三种情况不同元素个数不变，不同元素个数加一，不同元素个数加k，第一种情况就是原序列元素中从0到已给的最大元素这个区间里有元素空缺，那么且空缺元素加上最大元素值除2想上取整得到的元素在原序列已存在，同理剩下k-1次也是这样子重复插入同一个已存在的元素，而第二种就是原序列元素中从0到已给的最大元素这个区间里有元素空缺，那么且空缺元素加上最大元素值除2想上取整得到的元素在原序列不存在，同理剩下k-1次也是这样子重复插入同一个元素，即不同元素加1，最后一种情况就是从0到已给的最大元素这个区间里没有元素空缺，那么你去最小值就是已给的最大元素+1，然后的得到的结果就是已给的最大元素+1，然后k-1次重复这样的操作，每次都会插入不同的元素。

通过代码

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define sl(n) scanf("%lld",&n)
#define pl(n) printf("%lld",n)
#define pE printf("\n")
#define ull unsigned long long
#define pb push_back
#define pre(n) for(ll i=1;i<=n;i++)
#define rep(n) for(ll i=n;i>=1;i--)
#define pi pair<ll,ll>
int main()
{
 	ll t,i,j,n,k,m;
 	sl(t);
 	while(t--){
 		m=-1;
 		sl(n),sl(k);
 		map<ll,ll>a;
 		pre(n){
 			sl(j);
 			m=max(m,j);
			 a[j]++; 
		 }
		if(!k){
			pl(n);
			pE;
			continue;
		}
		bool flag=false ;
		ll temp ;
		for(i=0;i<=m;i++){
			if(a[i]==0){
				temp=i;
				flag = true;
				break;
			}
		}
		if(!flag){
			pl(n+k);
			pE;
			continue;
		}
		if(a[(temp+m+1)/2]==0)pl(n+1);
		else pl(n);
		pE;	
	 }
	return 0;
}