CF 1496B

最新推荐文章于 2024-10-15 19:28:18 发布

AMjieker

最新推荐文章于 2024-10-15 19:28:18 发布

阅读量210

点赞数

分类专栏： CF 文章标签：算法 c++ map

本文链接：https://blog.csdn.net/qq_51986723/article/details/115257299

版权

CF 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

本文介绍了在CF1496B题目中，如何通过巧妙利用map数据结构和max、mex概念，避免暴力模拟，解决大规模k值情况下的时间复杂度问题。通过分析max和mex的关系，作者提供了三种关键场景的解决方案，确保在最短时间找到正确答案。

摘要由CSDN通过智能技术生成

CF(穿越火线) 1496B

B. Max and Mex

这道题如果暴力模拟是会tle的，因为k非常大的(~~亲身经历~~)

但是我们可以发现max和mex大小关系不一样会造成结果不一致，很明显，
我们用map来存储一下下因为map会自排序
先排除一下k==0的情况，

case1: 如果mex > max 那么每一次结果都将稳步上升 $m e x = m a x + 1$ ; $m a x = r o u n d ((m a x + m e x) / 2)$
那么下一次 n也会增加

case2: 如果max > mex 但是 $r o u n d ((m a x + m e x) / 2)$ 得到的值是一个已有位置的值，那么下次mex和max将不会更新，然后他就卡bug了，跟k没关系总数是n

case3: 如果max > mex 但是 $r o u n d ((m a x + m e x) / 2)$ 得到的值是一个未有位置的值，那么总数将增加1 但是 mex + max 除以2的结果可能是mex 吗很显然不可能的所以下一次的mex 和max 也不会变所以总数是n+1

#include<iostream>
#include<map>
#include<cmath>
#define int long long
using namespace std;
signed main() {
	int t;cin>>t;
	while(t--){
		int n,k;cin>>n>>k;
		map<int,int> mp;
		int _max(0),_mex(0);
		for(int i=0;i<n;i++){
			int x;cin>>x;mp[x]++;
		} 
		if(k==0) {
			cout<<n<<endl;
			continue;
		}
		map<int,int>:: iterator it = mp.end();
		_max = ((--it)->first);
		
		for(int i=0;i<=n;i++) {
			if(mp.count(i)==0) {
				_mex=i;break;
			}
		}
		if(_max<_mex){
			cout<<n+k<<endl;
		} else {
			if(mp.count(round((_max+_mex)/2.0))){
				cout<<n<<endl;
			} else {
				cout<<n+1<<endl;
			}
		}
	}
	return 0;
}