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