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;
}