HDU - 4747 - Mex（线段树）

HDU - 4747 - Mex

用线段树维护 $mex(l,1),mex(l,2),...mex(l,n)$ 。
一开始可以预处理出 $l=1$ 的情况 。
然后删去 $a[1]$ ，由于 $mex$ 是递增的，因此只需要修改线段树上第一个比 $a[1]$ 大的值到下一个为 $1$ 的值这段区间改为 $a[1]$ 即可。
算法的复杂度是 $O(nlogn)$ 。

#include<bits/stdc++.h>
#define lson (rt<<1)
#define rson (rt<<1|1)
using namespace std;
typedef long long ll;
const int N=2e5+7;
ll sum[N<<2],ans;
int a[N],n,cur,lz[N<<2],nxt[N],mx[N<<2];
map<int,int> vis;
void push_up(int rt)
{
    sum[rt]=sum[lson]+sum[rson];
    mx[rt]=max(mx[lson],mx[rson]);
}
void push_down(int rt,int l,int r)
{
    if(lz[rt]!=-1)
    {
        lz[lson]=lz[rson]=lz[rt];
        mx[lson]=mx[rson]=lz[rt];
        int m=(l+r)>>1;
        sum[lson]=(ll)lz[lson]*(m-l+1);
        sum[rson]=(ll)lz[rson]*(r-m);
        lz[rt]=-1;
    }
}
void build(int rt,int l,int r)
{
    lz[rt]=-1;
    if(l==r)
    {
        if(a[l]<N) vis[a[l]]=1;
        while(vis.count(cur)) ++cur;
        mx[rt]=sum[rt]=cur;
        ans+=cur;
        return ;
    }
    int m=(l+r)>>1;
    build(lson,l,m);
    build(rson,m+1,r);
    push_up(rt);
}

void update(int rt,int l,int r,int ql,int qr,int v)
{
    if(ql<=l&&qr>=r)
    {
        lz[rt]=mx[rt]=v;
        sum[rt]=(ll)lz[rt]*(r-l+1);
        return ;
    }
    push_down(rt,l,r);
    int m=(l+r)>>1;
    if(ql<=m) update(lson,l,m,ql,qr,v);
    if(qr>m) update(rson,m+1,r,ql,qr,v);
    push_up(rt);
}

int query_l(int rt,int l,int r,int p)
{
    if(l==r) return mx[rt]>p?l:n+1;
    int m=(l+r)>>1,res;
    push_down(rt,l,r);
    if(mx[lson]>p) res=query_l(lson,l,m,p);
    else res=query_l(rson,m+1,r,p);
    push_up(rt);
    return res;
}

int main()
{
    while(~scanf("%d",&n))
    {
        if(n==0) break;
        vis.clear();
        for(int i=1;i<=n;++i) scanf("%d",&a[i]);
        for(int i=n;i>=1;--i)
        {
            if(!vis.count(a[i])) nxt[i]=n+1;
            else nxt[i]=vis[a[i]];
            vis[a[i]]=i;
        }
        ans=cur =0;
        vis.clear();
        build(1,1,n);
        for(int i=1;i<n;++i)
        {
            int l=query_l(1,1,n,a[i]);
            update(1,1,n,i,i,0);
            if(l<nxt[i]) update(1,1,n,l,nxt[i]-1,a[i]);
            ans+=sum[1];
        }
        printf("%I64d\n",ans);
    }
    return 0;
}