HDU

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;}