[BZOJ 2120]数颜色：带修改莫队

点击这里查看原题

与普通莫队相比，本题需要单点修改，因此需要分别记录查询和修改两种操作。
修改操作需要记录：

• pos:修改的位置
• x:修改前的值

查询操作需要记录：

• p:上一个修改操作的序号
• id:在输入数据中的顺序
• l,r:查询的区间

查询操作依然是按块排序，只不过每次查询时要处理好修改操作。
用一个vis数组记录该位置是否在目前的区间内，如果要修改一个在队中的位置，需要先出队，然后修改该位置的值，再入队。如果不在队中直接修改即可。

/*
User:Small
Language:C++
Problem No.:2120
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
const int M=1e4+5; 
int n,m,a[M],num[M*100],res,tot,cnt,pos[M],ans[M],b[M];
bool vis[M];
struct op1{
    int pos,x,y;
}ch[M];
struct op2{
    int pre,l,r,id;
}qu[M];
bool cmp(op2 a,op2 b){
    return pos[a.l]!=pos[b.l]?pos[a.l]<pos[b.l]:a.r<b.r;
}
void update(int x){
    if(vis[x]){
        num[a[x]]--;
        if(num[a[x]]==0) res--;
    }
    else{ 
        num[a[x]]++;
        if(num[a[x]]==1) res++;
    }
    vis[x]^=1;
}
void change(int x,int y){
    if(vis[x]){
        update(x);
        a[x]=y;
        update(x);
    }
    else a[x]=y;
}
void solve(){
    int l=1,r=0,now=0;
    for(int i=1;i<=cnt;i++){
        for(int j=now+1;j<=qu[i].pre;j++)
            change(ch[j].pos,ch[j].y);
        for(int j=now;j>qu[i].pre;j--)
            change(ch[j].pos,ch[j].x);
        for(int j=r+1;j<=qu[i].r;j++) update(j);
        for(int j=r;j>qu[i].r;j--) update(j);
        for(int j=l-1;j>=qu[i].l;j--) update(j);
        for(int j=l;j<qu[i].l;j++) update(j);
        ans[qu[i].id]=res;
        l=qu[i].l,r=qu[i].r,now=qu[i].pre;
    }
}
int main(){
    freopen("data.in","r",stdin);//
    scanf("%d%d",&n,&m);
    int t=sqrt(n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        b[i]=a[i];
        pos[i]=(i-1)/t+1;
    }
    for(int i=1;i<=m;i++){
        char op[5];
        int x,y;
        scanf("%s%d%d",op,&x,&y);
        if(op[0]=='R'){
            ch[++tot]=(op1){x,b[x],y};
            b[x]=y;
        }
        else qu[++cnt]=(op2){tot,x,y,cnt};
    }
    sort(qu+1,qu+cnt+1,cmp);
    solve();
    for(int i=1;i<=cnt;i++) printf("%d\n",ans[i]);
    return 0;
}