CF1628A-Meximum Array【二分】

正题

题目链接:http://codeforces.com/contest/1628/problem/A


题目大意

给出一个长度为\(n\)的序列\(a\)和一个空序列\(b\),你每次可以选择\(a\)的一个前缀,将它的\(mex\)加入序列\(b\)的末尾,然后将\(a\)的这个前缀删除。

求\(b\)的最大字典序。

\(1\leq \sum n\leq 2\times 10^5,0\leq a_i\leq n\)


解题思路

显然地我们每次肯定要选\(mex\)最大的一个前缀,为了最大化我们后面的值,那么我们显然是选的前缀越短越好。

而考虑如何快速求到这个前缀,我们可以考虑从小到大枚举这个\(mex\)的值,假设现在枚举到\(x\),那么证明我们得到的前缀\(1\sim r\)之中已经有\(0\sim x-2\)了,我们就先看这个前缀中是否有\(x-1\),如果有就枚举下一个数,不然我们就让\(r\)跳到第一个\(x\)处。

显然这样一次的复杂度是\(O(mex\log n)\)的,但是因为删除的数字个数和\(mex\)同级,而一个数字最多被删除一次,这样时间复杂度就是\(O(n\log n)\)了。

写个\(vector\)上二分就好了。


code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=2e5+10;
int T,n;vector<int> v[N],mex;
int lb(int x,int L){
	int l=0,r=v[x].size()-1;
	while(l<=r){
		int mid=(l+r)>>1;
		if(v[x][mid]<=L)l=mid+1;
		else r=mid-1;
	}
	return (l>=v[x].size())?(n+1):v[x][l];
}
int main()
{
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);mex.clear();
		for(int i=0;i<=n;i++)v[i].clear();
		for(int i=1,x;i<=n;i++){
			scanf("%d",&x);
			v[x].push_back(i);
		}
		int l=0;
		while(l<n){
			int r=l+1;
			for(int i=0;i<=n;i++){
				int p=lb(i,l);
				if(p>n){
					mex.push_back(i);
					break;
				}
				r=max(p,r);
			}
			l=r;
		}
		printf("%d\n",mex.size());
		for(int i=0;i<mex.size();i++)
			printf("%d ",mex[i]);
		putchar('\n');
	}
}
上一篇:选择排序算法


下一篇:双向链表