BZOJ 2120 数颜色

题目描述

墨墨购买了一套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支画笔中共有几种不同颜色的画笔。

样例输入

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

样例输出

4
4
3
4

提示

对于100%的数据，N≤10000，M≤10000，修改操作不多于1000次，所有的输入数据中出现的所有整数均大于等于1且不超过10^6。
2016.3.2新加数据两组by Nano_Ape

题解

计算出 i 的 前一个出现的位置
然后就和教主的魔法一样了
注意数组大小

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
int comp(int a, int b)
{
    return a > b;
}
using namespace std;
int pre[10005], front[1000005];
int in[10005];
int a[10005], b[10005];
int n, block;
void build(int x)
{
    for (int i = (x - 1) * block + 1; i <= min(x * block, n); i++)
    {
        pre[i] = b[i];
    }
    sort(pre + (x - 1) * block + 1, pre + min(x * block + 1, n + 1));
}
void rebuild(int k, int x)
{
    for (int i = 1; i <= n; i++)
        front[a[i]] = 0;
    a[k] = x;
    for (int i = 1; i <= n; i++)
    {
        int ls = b[i];
        b[i] = front[a[i]];
        if (ls != b[i])
            build(in[i]);
        front[a[i]] = i;
    }
}
int query(int c, int x)
{
    int l = (x - 1) * block + 1, r = min(n, x * block);
    int head = l;
    while (l <= r)
    {
        int m = (l + r) >> 1;
        if (pre[m] < c)
            l = m + 1;
        else
            r = m - 1;
    }
    return l - head;
}
int Query(int l, int r)
{
    int ans = 0;
    if (in[l] == in[r])
    {
        for (int i = l; i <= r; i++)
            if (b[i] < l)
                ans++;
    }
    else
    {
        for (int i = l; i <= min(block * in[l], n); i++)
            if (b[i] < l)
                ans++;
        for (int i = block * (in[r] - 1) + 1; i <= r; i++)
            if (b[i] < l)
                ans++;
    }
    for (int i = in [l] + 1; i < in[r]; i++)
        ans += query(l, i);
    return ans;
}
int main()
{
    //freopen("nt2011_color.in","r",stdin);
    //freopen("nt2011_color.out","w",stdout);
    //freopen("1.in", "r", stdin);
    int m;
    scanf("%d%d", &n, &m);
    block = sqrt(n);
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        b[i] = front[a[i]];
        front[a[i]] = i;
        in[i] = (i - 1) / block + 1;
    }
    int tot = in[n];
    for (int i = 1; i <= tot; i++)
        build(i);
    char s[10];
    int l, r;
    while (m--)
    {
        scanf("%s", s);
        if (s[0] == 'Q')
        {
            scanf("%d%d", &l, &r);
            printf("%d\n", Query(l, r));
        }
        else
        {
            scanf("%d%d", &l, &r);
            rebuild(l, r);
        }
    }
}