[Luogu5324][BJOI2019]删数(线段树)

CF风格题，先猜结论，记数列中i这个数共出现了cnt[i]次，那么所有区间[i-cnt[i]+1,i]的并集的补集大小就是答案。

于是我们只需要线段树维护每个位置是否被某个区间覆盖到即可，对于整体加减操作，设一个偏移量即可。

#include<cstdio>
#include<algorithm>
#define ls (x<<1)
#define rs (ls|1)
#define lson ls,L,mid
#define rson rs,mid+1,R
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
using namespace std;

const int N=600010;
int n,m,mx,py,L,p,x,a[N],num[N],v[N<<2],c[N<<2],mn[N<<2],tag[N<<2];

void upd(int x){
    if (mn[ls]<mn[rs]) mn[x]=mn[ls],c[x]=c[ls];
    else if (mn[ls]>mn[rs]) mn[x]=mn[rs],c[x]=c[rs];
        else mn[x]=mn[ls],c[x]=c[ls]+c[rs];
    if (!mn[x]) v[x]=c[x]; else v[x]=0;
}

void put(int x,int k){
    mn[x]+=k; tag[x]+=k;
    if (!mn[x]) v[x]=c[x]; else v[x]=0;
}

void push(int x){
    if (!tag[x]) return;
    put(ls,tag[x]); put(rs,tag[x]); tag[x]=0;
}

void build(int x,int L,int R){
    if (L==R){ c[x]=v[x]=1; return; }
    int mid=(L+R)>>1;
    build(lson); build(rson); upd(x);
}

void mdf(int x,int L,int R,int l,int r,int k){
    if (L==l && r==R){ put(x,k); return; }
    int mid=(L+R)>>1; push(x);
    if (r<=mid) mdf(lson,l,r,k);
    else if (l>mid) mdf(rson,l,r,k);
        else mdf(lson,l,mid,k),mdf(rson,mid+1,r,k);
    upd(x);
}

int que(int x,int L,int R,int l,int r){
    if (L==l && r==R) return v[x];
    int mid=(L+R)>>1; push(x);
    if (r<=mid) return que(lson,l,r);
    else if (l>mid) return que(rson,l,r);
        else return que(lson,l,mid)+que(rson,mid+1,r);
}

void Mdf(int x,int w){
    int k=num[py+x]+(w>0); num[py+x]+=w;
    if (x<=n) mdf(1,1,mx,py+x-k+1,py+x-k+1,w);
}

int main(){
    freopen("number.in","r",stdin);
    freopen("number.out","w",stdout);
    scanf("%d%d",&n,&m); mx=(n+m)*2+1; py=n+m; build(1,1,mx);
    rep(i,1,n) scanf("%d",&a[i]),Mdf(a[i],1);
    while (m--){
        scanf("%d%d",&p,&x);
        if (p>0) Mdf(a[p]+L,-1),a[p]=x-L,Mdf(a[p]+L,1);
        else if (x>0){
            py--; L++; int pos=py+n+1;
            if (num[pos]>0) mdf(1,1,mx,pos-num[pos]+1,pos,-1);
        }else{
            int pos=py+n+1; py++; L--;
            if (num[pos]>0) mdf(1,1,mx,pos-num[pos]+1,pos,1);
        }
        printf("%d\n",que(1,1,mx,py+1,py+n));
    }
    return 0;
}