CF1436E Complicated Computations(权值线段树)
题意:
给一个长度为n的数组,询问这个数组的所有子串的MEX组成的序列的MEX
题解:
权值线段树,枚举每个数每个出现的位置和上次出现的位置之间的区间,看看这个区间的1到a[i]-1是否全部出现,这个只要维护每个数当前最后出现的位置,然后看看是否有数的位置小于上次出现的位置就好。
这一步用权值线段树就能完成。
//求一个序列的所有子序列的MEX组成的序列的MEX #include<bits/stdc++.h> using namespace std; const int maxn=1e5+100; int a[maxn]; int pre[maxn]; int ans[maxn]; int n; struct node { int l,r,sum; }segTree[maxn<<2]; void build (int i,int l,int r) { segTree[i].l=l; segTree[i].r=r; if (l==r) { segTree[i].sum=0; return; } int mid=(l+r)>>1; build(i<<1,l,mid); build(i<<1|1,mid+1,r); segTree[i].sum=min(segTree[i<<1].sum,segTree[i<<1|1].sum); } void up (int i,int p,int v) { if (segTree[i].l==p&&segTree[i].r==p) { segTree[i].sum=v; return; } int mid=(segTree[i].l+segTree[i].r)>>1; if (p<=mid) up(i<<1,p,v); if (p>mid) up(i<<1|1,p,v); segTree[i].sum=min(segTree[i<<1].sum,segTree[i<<1|1].sum); } int query (int i,int l,int r) { if (segTree[i].l>=l&&segTree[i].r<=r) { return segTree[i].sum; } int mid=(segTree[i].l+segTree[i].r)>>1; int ans=1e9; if (l<=mid) ans=min(ans,query(i<<1,l,r)); if (r>mid) ans=min(ans,query(i<<1|1,l,r)); return ans; } int main () { scanf("%d",&n); for (int i=1;i<=n;i++) scanf("%d",a+i),ans[i]=1;ans[n+1]=1;ans[n+2]=1; for (int i=1;i<=n;i++) if (a[i]!=1) ans[1]=0; build(1,1,n); for (int i=1;i<=n;i++) { if (a[i]==1) { up(1,a[i],i); pre[a[i]]=i; continue; } int p=query(1,1,a[i]-1);//找到1到a[i]-1这个区间的最小位置,没出现就是0 //printf("%d\n",p); if (p>pre[a[i]]) ans[a[i]]=0; up(1,a[i],i); pre[a[i]]=i; } for (int i=2;i<=n+1;i++) { int p=query(1,1,i-1); if (p>pre[i]) ans[i]=0;//如果有最小的小于pre[i] } for (int i=1;i<=n+2;i++) { if (ans[i]) { printf("%d\n",i); break; } } }