HYSBZ - 2120 数颜色(带修改的莫队)

墨墨购买了一套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
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e4 + 5;
int block_len;
struct node {
    int l, r, bl, br, id, t;
    node() {}
    node(int l, int r, int id, int t) : l(l), r(r), id(id), t(t) {
        bl = l / block_len;
        br = r / block_len;
    }
    bool operator<(const node &x) const {
        if (bl == x.bl) {
            if (br == x.br)
                return t < x.t;
            else
                return r < x.r;
        } else
            return bl < x.bl;
    }
} query[maxn];
struct node2 {
    int pos, pre, cur;
} c[maxn];
int m, n, Q, T, ans;
int a[maxn], now[maxn], cnt[maxn * 100], sum[maxn];
void add(int val) {
    if (cnt[val] == 0) ans++;
    cnt[val]++;
}
void del(int val) {
    cnt[val]--;
    if (cnt[val] == 0) ans--;
}
void work(int l, int r, int pos, int val) {
    if (pos >= l && pos <= r) {
        del(a[pos]);
        add(val);
    }
    a[pos] = val;
}
int main() {
    scanf("%d%d", &n, &m);
    block_len = pow(n, 2.0 / 3.0);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        now[i] = a[i];
    }
    char op[3];
    for (int i = 1; i <= m; i++) {
        scanf("%s", op);
        if (op[0] == 'Q') {
            int l, r;
            scanf("%d%d", &l, &r);
            query[++Q] = node(l, r, Q, T);
        } else {
            int pos, x;
            scanf("%d%d", &pos, &x);
            c[++T] = {pos, now[pos], x};
            now[pos] = x;
        }
    }
    sort(query + 1, query + Q + 1);
    int l = 1, r = 1, curT = 0;
    ans = 1;
    cnt[a[1]]++;
    for (int i = 1; i <= Q; i++) {
        node &qr = query[i];
        while (r < qr.r) add(a[++r]);
        while (l > qr.l) add(a[--l]);
        while (r > qr.r) del(a[r--]);
        while (l < qr.l) del(a[l++]);
        while (curT < qr.t) curT++, work(l, r, c[curT].pos, c[curT].cur);
        while (curT > qr.t) work(l, r, c[curT].pos, c[curT].pre), curT--;
        sum[qr.id] = ans;
        // printf("R%d %d\n",qr.id,curT);
        // for(int j=1;j<=n;j++) printf("%d ",a[j]);puts("");
    }
    for (int i = 1; i <= Q; i++) printf("%d\n", sum[i]);
    // system("pause");
    return 0;
}

猜你喜欢

转载自blog.csdn.net/blue_kid/article/details/79451157