BZOJ2120 数颜色 【带修莫队】

BZOJ2120 数颜色

---

Description

墨墨购买了一套N支彩色画笔（其中有些颜色可能相同），摆成一排，你需要回答墨墨的提问。墨墨会像你发布如下指令：
1、 Q L R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。
2、 R P Col 把第P支画笔替换为颜色Col。
为了满足墨墨的要求，你知道你需要干什么了吗？

Input

第1行两个整数N，M，分别代表初始画笔的数量以及墨墨会做的事情的个数。第2行N个整数，分别代表初始画笔排中第i支画笔的颜色。第3行到第2+M行，每行分别代表墨墨会做的一件事情，格式见题干部分。

Output

对于每一个Query的询问，你需要在对应的行中给出一个数字，代表第L支画笔到第R支画笔中共有几种不同颜色的画笔。

Sample Input

6 5
1 2 3 4 5 5
Q 1 4
Q 2 6
R 1 2
Q 1 4
Q 2 6

Sample Output

4
4
3
4

HINT

对于100%的数据，N≤10000，M≤10000，修改操作不多于1000次，所有的输入数据中出现的所有整数均大于等于1且不超过10^6。

---

挺好的一道带修改莫队入门题

一开始没有考虑在修改的基础上进行累加，不能一直从原状态上进行累加，因为时间是固定的，所以我们可以保证修改一定是有序的

就是代码中的last数组

其他没什么好说的，但是提醒一下，虽然sqrt（n）也可以过，但是siz取n的三分之二次方最优

---

#include<bits/stdc++.h>
using namespace std;
#define N 1000010
struct Node{int pos,x,y;}t[N];
struct Ask{int l,r,id,tim;}p[N];
int block[N],a[N],last[N],ans[N],cnt[N];
int n,m,totp=0,tott=0;
int l=1,r=0,tmp=0,ansnow=0;
bool cmp1(Ask a,Ask b){
    if(block[a.l]==block[b.l])return a.r<b.r;
    else return block[a.l]<block[b.l];
}
void modify(int pos,int y){
    if(l<=pos&&pos<=r){
        if(--cnt[a[pos]]==0)ansnow--;
        a[pos]=y;
        if(++cnt[a[pos]]==1)ansnow++;
    }else a[pos]=y;
}
void update(int pos,int typ){
    if(typ==1)ansnow+=(++cnt[a[pos]]==1);
    else ansnow-=(--cnt[a[pos]]==0);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),last[i]=a[i];
    for(int i=1;i<=m;i++){
        char c[5];int x,y;
        scanf("%s%d%d",c,&x,&y);
        if(c[0]=='Q')p[++totp]=(Ask){x,y,totp,tott};
        else{
            t[++tott]=(Node){x,last[x],y};
            last[x]=y;
        }
    }
    int siz=sqrt(n);
    for(int i=1;i<=n;i++)block[i]=(i-1)/siz+1;
    sort(p+1,p+totp+1,cmp1);
    for(int i=1;i<=totp;i++){
        while(tmp<p[i].tim)++tmp,modify(t[tmp].pos,t[tmp].y);
        while(tmp>p[i].tim)modify(t[tmp].pos,t[tmp].x),tmp--;
        while(l<p[i].l)update(l++,-1);
        while(l>p[i].l)update(--l,1);
        while(r<p[i].r)update(++r,1);
        while(r>p[i].r)update(r--,-1);
        ans[p[i].id]=ansnow;
    }
    for(int i=1;i<=totp;i++)printf("%d\n",ans[i]);
    return 0;
}