P1903 [国家集训队]数颜色 / 维护队列 (带修莫队)

题目描述

墨墨购买了一套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。

本题可能轻微卡常数

因为某些nt操作导致改了一天的bug, 所以记录一下 (不仔细读题的下场

排序条件尽量用三元表达式, 避免使用if else if, 后者花的时间很可能是不可接受的

举个例子, 上面是三元表达式排序, 下面是if else if判断排序, 时间差距甚至达到了一倍甚至更多

 

#include <iostream>
#include <algorithm>
#include <climits>
#include <cmath>
using namespace std;
constexpr int MAXN = 2e6 + 7, MAXM = 2e6 + 7, inf = 0x3f3f3f3f;
typedef long long ll;
int Block_Size, n, m;
int a[MAXN], id[MAXN], color[MAXN], ans[MAXN], res, qc, mc, ls = 1, rs;

inline int read()
{
    char c = getchar();
    int x = 0, f = 1;
    while (c < '0' || c > '9')
    {
        if (c == '-')
            f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9')
    {
        x = x * 10 + c - '0', c = getchar();
    }
    return x * f;
}
struct query
{
    int l;
    int r;
    int qid;
    int time;
    query(int l = 0, int r = 0, int qid = 0, int time = 0) : l(l), r(r), qid(qid), time(time)
    {
    }

    bool operator<(const query &other) const
    {
        //一定要避免使用if else if的形式!!!!
        //例如下面的写法

        return id[l] == id[other.l] ? id[r] == id[other.r] ? time < other.time : r < other.r : l < other.l;
        // if (id[l] == id[other.l])
        // {
        //     if (id[r] == id[other.r])
        //         return l < other.l;
        //     return r < other.r;
        // }
        // return time < other.time;
    }
} q[MAXM];

struct modify
{
    int pos;
    int val;
    modify(int pos = 0, int val = 0) : pos(pos), val(val)
    {
    }
} mod[MAXM];

void del(int val)
{
    if (--color[val] == 0)
        --res;
}

void add(int val)
{
    if (++color[val] == 1)
        ++res;
}

void update(int x, int i)
{
    if (q[i].l <= mod[x].pos && mod[x].pos <= q[i].r)
    {
        del(a[mod[x].pos]);
        add(mod[x].val);
    }
    swap(a[mod[x].pos], mod[x].val);
}

int main()
{
    char op;
    int l, r, now = 0;
    n = read();
    m = read();
    Block_Size = pow(n, 2.0 / 3.0);
    for (int i = 1; i <= n; ++i)
    {
        a[i] = read();
        id[i] = (i - 1) / Block_Size;
    }

    for (int i = 1; i <= m; ++i)
    {
        scanf("%c", &op);
        l = read();
        r = read();
        if (op == 'R')
        {
            ++mc;
            mod[mc].pos = l;
            mod[mc].val = r;
        }
        else
        {
            ++qc;
            q[qc].l = l;
            q[qc].r = r;
            q[qc].time = mc;
            q[qc].qid = qc;
        }
    }
    sort(q + 1, q + qc + 1);

    for (int i = 1; i <= qc; ++i)
    {
        while (ls < q[i].l)
            del(a[ls++]);
        while (ls > q[i].l)
            add(a[--ls]);

        while (rs < q[i].r)
            add(a[++rs]);
        while (rs > q[i].r)
            del(a[rs--]);

        while (now < q[i].time)
            update(++now, i);
        while (now > q[i].time)
            update(now--, i);

        ans[q[i].qid] = res;
    }

    for (int i = 1; i <= qc; ++i)
    {
        printf("%d
", ans[i]);
    }

    return 0;
}