[bzoj2120] 数颜色

[bzoj2120] 数颜色

---

带修改莫队，块的大小是n^(2/3)，这样我们可以保证复杂度在n^(5/3)次方。

• 代码

#include<bits/stdc++.h>
using namespace std;
const int N=5e4+4;
const int blksz=1357;
inline int blk(int x) {
    return x/blksz;
}
int col[N],colcur[N],n,m;
struct question {
    int l,r,t;
    bool operator<(const question b)const {
        if(blk(l)^blk(b.l))return l<b.l;
        if(blk(r)^blk(b.r))return r<b.r;
        return t<b.t;
    }
} q[N];
int cnt1;
struct modify {
    int pos,col,colback,t;
} opt[N];
int cnt2;
int c[4000010];
int lstt,lstl,lstr,tpos;
int ans=0;
inline void add(int col,int sz) {
    c[col]+=sz;
    if(c[col]==1&&sz>0)ans++;
    if(c[col]==0&&sz<0)ans--;
}
inline int getopt() {
    char x=0;
    while(x<'A'||x>'Z')x=getchar();
    return x=='Q'?1:0;
}
int ret[N];
int main() {
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++) {
        scanf("%d",&col[i]);
        colcur[i]=col[i];
    }
    for(int i=1; i<=m; i++) {
        ret[i]=-1;
        int t;
        t=getopt();
        if(t==1) {
            q[++cnt1].t=i;
            scanf("%d%d",&q[cnt1].l,&q[cnt1].r);
        } else {
            opt[++cnt2].t=i;
            scanf("%d",&opt[cnt2].pos);
            opt[cnt2].colback=colcur[opt[cnt2].pos];
            scanf("%d",&opt[cnt2].col);
            colcur[opt[cnt2].pos]=opt[cnt2].col;
        }
    }
    sort(q+1,q+cnt1+1);
    for(int id=1; id<=cnt1; id++) {
        if(q[id].t>lstt) {
            while(tpos<cnt2&&opt[tpos+1].t<=q[id].t) {
                ++tpos;
                int pos=opt[tpos].pos,cc=opt[tpos].col;
                if(pos>=lstl&&pos<=lstr)
                    add(col[pos],-1),add(cc,1);
                col[pos]=cc;
            }
        }
        if(q[id].t<lstt) {
            while(tpos&&opt[tpos].t>q[id].t) {
                int pos=opt[tpos].pos,cc=opt[tpos].colback;
                if(pos>=lstl&&pos<=lstr)
                    add(col[pos],-1),add(cc,1);
                col[pos]=cc;
                --tpos;
            }
        }
        lstt=q[id].t;
        if(q[id].r>lstr)
            for(int i=lstr+1; i<=q[id].r; i++)
                add(col[i],1);
        if(q[id].l<lstl)
            for(int i=lstl-1; i>=q[id].l; i--)
                add(col[i],1);
        if(q[id].r<lstr)
            for(int i=lstr; i>q[id].r; i--)
                add(col[i],-1);
        if(q[id].l>lstl)
            for(int i=lstl; i<q[id].l; i++)
                add(col[i],-1);
        ret[q[id].t]=ans;
        lstl=q[id].l,lstr=q[id].r;
    }
    for(int i=1; i<=m; i++) {
        if(~ret[i]) {
            printf("%d\n",ret[i]);
        }
    }
}