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
",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
",i);
            break;
        }
    }
} 
原文地址:https://www.cnblogs.com/zhanglichen/p/13892659.html