洛谷 P1903 数颜色 / 维护队列(带修改莫队+经典)

题目链接：

https://www.luogu.org/problemnew/show/P1903

题意：

给你一个含n个整数的序列，每个数代表一种颜色，给你若干个查询操作和修改操作，查询即查询$[l,r]$区间内颜色数，修改即修改$a[x]$的颜色为$val$，对于每个查询输出颜色数；

分析：

带修改莫队；
(先放代码，下次再补)

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>

using namespace std;

#define  inf 0x7f7f7f7f
#define  maxn 50050
#define  N 2000100
#define  P 2

typedef long long ll;
typedef struct {
    int u, v, next, w;
} Edge;
Edge e[P];
int cnt, head[P];

inline void add1(int u, int v) {
    e[cnt].u = u;
    e[cnt].v = v;
    //e[cnt].w=w;
    // e[cnt].f=f;
    e[cnt].next = head[u];
    head[u] = cnt++;
    //e[cnt].u=v;
//    e[cnt].v=u;
//    e[cnt].w=0;
//    e[cnt].f=-f;
//    e[cnt].next=head[v];
//    head[v]=cnt++;
}

inline void write(int x) {
    if (x < 0)
        putchar('-'), x = -x;
    if (x > 9)
        write(x / 10);
    putchar(x % 10 + '0');
}

inline int read() {
    int x = 0, f = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-')
            f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        x = x * 10 + c - '0';
        c = getchar();
    }
    return x * f;
}
struct node1{
    int l,r,id,ti;
}p[maxn];
struct node2{
    int x,val;
}q[maxn];
char c[5];
int a[N],res,n,m,num1,num2,ans[maxn],pos[maxn],cot[N];

bool com(node1 a,node1 b){
    if(pos[a.l]!=pos[b.l])return pos[a.l]<pos[b.l];
    else if(pos[a.r]!=pos[b.r])return pos[a.r]<pos[b.r];
    else return a.ti<b.ti;
}
inline void del(int x){
    if(cot[a[x]]==1)res--;
    cot[a[x]]--;
}
inline void add(int x){
    if(cot[a[x]]==0)res++;
    cot[a[x]]++;
}
inline void change(int x,int i){
    if(q[x].x>=p[i].l&&q[x].x<=p[i].r){
        if(cot[a[q[x].x]]==1)res--;
        cot[a[q[x].x]]--;
        if(cot[q[x].val]==0)res++;
        cot[q[x].val]++;
    }
    swap(a[q[x].x],q[x].val);//it's so amazing!
}
int main() {
    cin>>n>>m;
    int sz=pow(n,2.0/3.0);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        pos[i]=i/sz;
    }
    for(int i=1;i<=m;i++){
        scanf("%s",c);
        if(c[0]=='Q'){
            num1++;
            scanf("%d%d",&p[num1].l,&p[num1].r);
            p[num1].id=num1,p[num1].ti=num2;
        }
        else if(c[0]=='R'){
            num2++;
            scanf("%d%d",&q[num2].x,&q[num2].val);
        }
    }
    sort(p+1,p+1+num1,com);
    int now=0,r=0,l=1;
    for(int i=1;i<=num1;i++){
        while(l<p[i].l){ del(l);l++; }
        while(l>p[i].l){ l--;add(l); }
        while(r<p[i].r){ r++;add(r); }
        while(r>p[i].r){ del(r);r--; }
        while(now<p[i].ti){change(++now,i);}
        while(now>p[i].ti){change(now--,i);}
        ans[p[i].id]=res;
    }
    for(int i=1;i<=num1;i++){
        cout<<ans[i]<<endl;
    }
    return 0;
}

我们坚持一件事情，并不是因为这样做了会有效果，而是坚信，这样做是对的。
——哈维尔