洛谷P1903 [国家集训队]数颜色 / 维护队列 [带修改莫队]

传送门
题意：有一个长度为n的数组，q次操作，询问[L,R]区间不同数字个数或者修改某个位置的数字

题解：带修改莫队模板题。比普通的莫队题多了个修改操作，带修改莫队也就比普通莫队多了一维时间，记录之前一次修改操作的位置，具体实现看代码

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
#define debug(x) cout<<#x<<" is "<<x<<endl;

const int maxn=1e6+5;

struct ac1{
    int id;
    int L;
    int R;
    int pre;
}q[maxn];

struct ac2{
    int pos;
    int col;
}cg[maxn];

int a[maxn],in[maxn],len,ans,acc[maxn];

bool cmp(struct ac1 aa,struct ac1 bb){
    if((aa.L/len+(aa.L%len!=0))!=(bb.L/len+(bb.L%len!=0)))return (aa.L/len+(aa.L%len!=0))<(bb.L/len+(bb.L%len!=0));
    else if((aa.R/len+(aa.R%len!=0))!=(bb.R/len+(bb.R%len!=0)))return (aa.R/len+(aa.R%len!=0))<(bb.R/len+(bb.R%len!=0));
    return aa.pre<bb.pre;
}

void add(int pos,int val){
    in[a[pos]]+=val;
    if(in[a[pos]]==1&&val>0)ans++;
    if(in[a[pos]]==0&&val<0)ans--;
}

void chg(int num,int i){
    if(cg[num].pos>=q[i].L&&cg[num].pos<=q[i].R){
        in[a[cg[num].pos]]--;
        in[cg[num].col]++;
        if(in[a[cg[num].pos]]==0)ans--;
        if(in[cg[num].col]==1)ans++;
    }
    swap(a[cg[num].pos],cg[num].col);
}

int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    int id1,id2;
    id1=id2=0;
    for(int i=1;i<=m;i++){
        char ch[5];
        scanf("%s",ch);
        if(ch[0]=='Q'){
            ++id1;
            q[id1].id=id1;
            scanf("%d%d",&q[id1].L,&q[id1].R);
            q[id1].pre=id2;
           // debug(id2);
        }
        else{
            ++id2;
            scanf("%d%d",&cg[id2].pos,&cg[id2].col);
        }
    }
    len=pow(n,0.67);
    sort(q+1,q+1+id1,cmp);
    int l,r,tim;
    l=r=tim=0;
    for(int i=1;i<=id1;i++){
        while(l>q[i].L){
            add(--l,1);
        }
        while(r<q[i].R){
            add(++r,1);
        }
        while(l<q[i].L){
            add(l++,-1);
        }
        while(r>q[i].R){
            add(r--,-1);
        }
        while(tim>q[i].pre){
            chg(tim--,i);
        }
        while(tim<q[i].pre){
            chg(++tim,i);
        }
        acc[q[i].id]=ans;
    }
    for(int i=1;i<=id1;i++)printf("%d\n",acc[i]);
    return 0;
}