2120: 数颜色（带修改莫队）

2120: 数颜色

Time Limit: 6 Sec  Memory Limit: 259 MB
Submit: 7599  Solved: 3087
[Submit][Status][Discuss]

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。

2016.3.2新加数据两组by Nano_Ape

 

分析

待修改莫队。

与普通莫队相比较，多了修改操作，那么对于每一个查询[l,r]，再记录一个t值，表示当前询问最近的一次修改的编号，即询问前，要操作t次。

排序时t为第三关键字。

然后维护三个指针L,R,T。与普通莫队不同的是：对于当前询问，t>T，那么依次更新t+1,t+2...T次操作；对于t<T，那么减去T,T-1...t。

code

#include<cstdio>
#include<algorithm>
#include<cmath>

using namespace std;

const int N = 100100;

struct Query{
    int l,r,id,t,k; 
    //依次表示 查询的左端点，右端点，查询序号，查询前的最近一次修改，左端点属于的块 
}q[N];
struct Change{
    int p,x,y;
    //依次表示 修改的位置，修改后的值，修改前的值（原值） 
}c[N];
int last[N],ans[N],cnt[1000100],v[N];
int C,D,Ans,L,R,T,n,m;
char s[10];

bool cmp(Query a,Query b) {
    if (a.k != b.k) return a.k < b.k;
    if (a.r != b.r) return a.r < b.r;
    return a.t < b.t;
}
void add(int x) {
    cnt[x] ++;
    if (cnt[x] == 1) Ans ++;
}
void del(int x) {
    cnt[x] --;
    if (cnt[x] == 0) Ans --;
}
void update(int p,int x) { // 将p位置修改为x 
    if (L <= p && p <= R) del(v[p]),add(x);
    v[p] = x;
}
void work() {
    L = 1,R = 0,T = 0;
    for (int i=1; i<=C; ++i) {
        while (T < q[i].t) update(c[T+1].p,c[T+1].x),T++;
        while (T > q[i].t) update(c[T].p,c[T].y),T--;
        
        while (L < q[i].l) del(v[L]),L++;
        while (L > q[i].l) L--,add(v[L]);
        while (R < q[i].r) R++,add(v[R]);
        while (R > q[i].r) del(v[R]),R--;

        ans[q[i].id] = Ans;
    }
}
int main() {
    scanf("%d%d",&n,&m);
    int B = pow(double(n),2.0/3.0);
    for (int i=1; i<=n; ++i) 
        scanf("%d",&v[i]),last[i] = v[i];
    for (int a,b,i=1; i<=m; ++i) {
        scanf("%s%d%d",s,&a,&b);
        if (s[0]=='Q') {
            q[++C].l = a;q[C].r = b;q[C].t = D;q[C].k = (n-1)/B+1;q[C].id = C;
        }
        else {
            c[++D].p = a;c[D].x = b;c[D].y = last[a];last[a] = b;
        }
    }
    sort(q+1,q+C+1,cmp);
    work();
    for (int i=1; i<=C; ++i) 
        printf("%d\n",ans[i]);
    return 0;
}