[BZOJ 2120]数颜色

Posted 2020-10-18

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。

题解

带修改的莫队，和一般的莫队差不多，就是在操作前对“修改”进行操作。

 1 //It is made by Awson on 2017.12.24
 2 #include <map>
 3 #include <set>
 4 #include <cmath>
 5 #include <ctime>
 6 #include <queue>
 7 #include <stack>
 8 #include <vector>
 9 #include <cstdio>
10 #include <string>
11 #include <cstdlib>
12 #include <cstring>
13 #include <iostream>
14 #include <algorithm>
15 #define LL long long
16 #define Max(a, b) ((a) > (b) ? (a) : (b))
17 #define Min(a, b) ((a) < (b) ? (a) : (b))
18 using namespace std;
19 const int N = 10000;
20 
21 int n, m, lim, a[N+5], x, y;
22 char ch[5];
23 struct Query {
24     int l, r, pre, id;
25     bool operator < (const Query &b) const {
26     return l/lim == b.l/lim ? r < b.r : l < b.l;
27     }
28 }q[N+5];
29 struct Opt {
30     int x, y, pre;
31 }p[N+5];
32 int path[N+5], cntp, cntq, kp[N+5];
33 int cnt[N*100+5];
34 
35 void update(int x, int color, int l, int r, int &ans) {
36     if (l <= x && x <= r) {
37     ans -= (--cnt[a[x]] == 0);
38     ans += (++cnt[a[x] = color] == 1);
39     }else a[x] = color;
40 }
41 void work() {
42     scanf("%d%d", &n, &m); lim = sqrt(n);
43     for (int i = 1; i <= n; i++) scanf("%d", &a[i]), path[i] = a[i];
44     for (int i = 1; i <= m; i++) {
45     scanf("%s%d%d", ch, &x, &y);
46     if (ch[0] == ‘R‘) {
47         p[++cntp].x = x, p[cntp].y = y, p[cntp].pre = path[x];
48         path[x] = y;
49     }else {
50         q[++cntq].l = x, q[cntq].r = y, q[cntq].pre = cntp, q[cntq].id = cntq;
51     }
52     }
53     sort(q+1, q+cntq+1);
54     int curp = 0, curl = 1, curr = 0, ans = 0, l, r, rp;
55     for (int i = 1; i <= cntq; i++) {
56     l = q[i].l, r = q[i].r, rp = q[i].pre;
57     while (curp < rp) curp++, update(p[curp].x, p[curp].y, curl, curr, ans);
58     while (curp > rp) update(p[curp].x, p[curp].pre, curl, curr, ans), curp--;
59     while (curl < l) ans -= (--cnt[a[curl++]] == 0);
60     while (curl > l) ans += (++cnt[a[--curl]] == 1);
61     while (curr < r) ans += (++cnt[a[++curr]] == 1);
62     while (curr > r) ans -= (--cnt[a[curr--]] == 0);
63     kp[q[i].id] = ans;
64     }
65     for (int i = 1; i <= cntq; i++) printf("%d\n", kp[i]);
66 }
67 int main() {
68     work();
69     return 0;
70 }