带修改的莫队

https://www.luogu.org/problem/P1903

后台数据太强了，只得了60分

题目描述

墨墨购买了一套N支彩色画笔（其中有些颜色可能相同），摆成一排，你需要回答墨墨的提问。墨墨会向你发布如下指令：

1、 Q L R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。

2、 R P Col 把第P支画笔替换为颜色Col。

为了满足墨墨的要求，你知道你需要干什么了吗？

输入格式

第1行两个整数N，M，分别代表初始画笔的数量以及墨墨会做的事情的个数。

第2行N个整数，分别代表初始画笔排中第i支画笔的颜色。

第3行到第2+M行，每行分别代表墨墨会做的一件事情，格式见题干部分。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

4
4
3
4

说明/提示

对于30%的数据，n,m \leq 10000n,m≤10000

对于60%的数据，n,m \leq 50000n,m≤50000

对于所有数据，n,m \leq 133333n,m≤133333

所有的输入数据中出现的所有整数均大于等于1且不超过10^6。

本题可能轻微卡常数

来源：bzoj2120

本题数据为洛谷自造数据，使用CYaRon耗时5分钟完成数据制作。

 

带修改的莫队需要在原来莫队的基础上加上个时间指针，用来判断当前的状态是修改了几次后得到的，然后再修改。 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e6+7;
ll arr[N],cnt[N];
ll ans=0,base;
ll anss[N];
struct stu{
    ll l,r,k,id,p;
    bool friend operator < (const stu &x,const stu &y){
        if(x.k!=y.k) return x.l<y.l;
        if(x.r!=y.r) return x.r<y.r; 
        return x.p<y.p;
    }
}node[N];
struct stuu{
    ll pos,color;
}a[N];
void change(ll now,ll i){
    if(node[i].l<=a[now].pos&&node[i].r>=a[now].pos){//只要修改的数据在查询的范围内，才会对ans造成影响 
        if(--cnt[arr[a[now].pos]]==0) ans--;
        if(cnt[a[now].color]++==0) ans++;
    }
    swap(a[now].color,arr[a[now].pos]);//修改，（交换，为下次返回做好铺垫） 
}
void add(ll x){
    cnt[arr[x]]++;
    if(cnt[arr[x]]==1) ans++;
}
void del(ll x){
    cnt[arr[x]]--;
    if(cnt[arr[x]]==0) ans--;
}
int main(){
    ll n,m;
    scanf("%lld%lld",&n,&m);
    base =sqrt(n);
    for(int i=1;i<=n;i++) scanf("%lld",&arr[i]);
    ll pos=0,pos2=0;
    for(int i=1;i<=m;i++){
        char judge[N];
        scanf("%s",judge);
        if(judge[0]=='Q'){
            ll l,r;
            scanf("%lld%lld",&l,&r);
            ++pos;
            node[pos].l=l,node[pos].r=r;
            node[pos].id=pos;
            node[pos].p=pos2;//到目前位置修改的次数，同时记录的也是上次修该的位置 
        }
        else {
            ll p,q;
            scanf("%lld%lld",&p,&q);
            ++pos2;
            a[pos2].pos=p;
            a[pos2].color=q;
        }
    }
    sort(node+1,node+1+pos);
    ll l=1,r=0,now=0;
    for(int i=1;i<=pos;i++){
        ll ql=node[i].l;
        ll qr=node[i].r;
        while(l<ql) del(l++);
        while(l>ql) add(--l);
        while(r>qr) del(r--);
        while(r<qr) add(++r);
        
        while(now<node[i].p) change(++now,i);//判断当前修改的次数，如果小于第i次查询要求的次数，那么增加 
        while(now>node[i].p) change(now--,i);
        anss[node[i].id]=ans;
    }
    for(int i=1;i<=pos;i++) printf("%lld\n",anss[i]);
    return 0;
}