CF1436E Complicated Computations
题面传送门
首先考虑一个结论:将一个区间中的数去重后这个区间\(mex\)没有影响。
所以我们只要求出那些\(mex\)中有几个数。
一个区间\(mex=k\)的条件是区间没有\(k\)且区间有\(1\)到\(k-1\)。
考虑对于同样的数分段。
那么对于每个段这样查询即可。
即满足两个条件:\([l\leq i \leq r]\)和所有不同的\(1\sim k-1\)种类等于\(k-1\)。
如果直接像\(HH的项链\)那样那就要二维树状数组维护要两个\(log\)。
转化一下,则就是判断\(1 \sim k-1\)中最小的出现的位置是否小于\(l\)。
那么就可以复杂度\(O(nlogn)\)。
代码实现:
#include<cstdio>
#include<vector>
#include<cstring>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
int n,m,k,x,y,z,ans[100039],a[100039],pus,maxn;
int last[100039],f[400039],g[100039];
inline void get(int x,int z,int l,int r,int now){
if(l==r){f[now]=z;return;}
int m=(l+r)>>1;
if(x<=m) get(x,z,l,m,now<<1);
else get(x,z,m+1,r,now<<1|1);
f[now]=min(f[now<<1],f[now<<1|1]);
}
inline int find(int x,int y,int l,int r,int now){
if(x<=l&&r<=y) return f[now];
int m=(l+r)>>1,fs1=1e9,fs2=1e9;
if(x<=m) fs1=find(x,y,l,m,now<<1);
if(y>m)fs2=find(x,y,m+1,r,now<<1|1);
return min(fs1,fs2);
}
int main(){
// freopen("1.in","r",stdin);
register int i;
scanf("%d",&n);
for(i=1;i<=n;i++) scanf("%d",&a[i]),maxn=max(maxn,a[i]),ans[1]|=(a[i]!=1?1:0),ans[2]|=(a[i]==1?1:0);
for(i=1;i<=n;i++){
g[i]=last[a[i]];
last[a[i]]=i;
}
for(i=1;i<=n;i++){
get(a[i],i,1,n+1,1);
if(a[i]!=1){
pus=find(1,a[i]-1,1,n+1,1);
//printf("%d %d\n",a[i],pus);
if(pus>g[i]) ans[a[i]]=1;
}
}
for(i=2;i<=maxn+1;i++){
// printf("%d %d\n",i,find(1,i-1,1,n+1,1));
if(find(1,i-1,1,n+1,1)>last[i]) ans[i]=1;
}
for(i=1;i<=n+2;i++) if(!ans[i]){printf("%d\n",i);return 0;}
}