【BZOJ】2453: 维护队列&&【BZOJ】2120: 数颜色 二分+分块 双倍经验

Description

你小时候玩过弹珠吗？

小朋友A有一些弹珠，A喜欢把它们排成队列，从左到右编号为1到N。为了整个队列鲜艳美观，小朋友想知道某一段连续弹珠中，不同颜色的弹珠有多少。当然，A有时候会依据个人喜好，替换队列中某个弹珠的颜色。但是A还没有学过编程，且觉得头脑风暴太浪费脑力了，所以向你来寻求帮助。

Input

输入文件第一行包含两个整数N和M。

第二行N个整数，表示初始队列中弹珠的颜色。

接下来M行，每行的形式为“Q L R”或“R x c”，“Q L R”表示A想知道从队列第L个弹珠到第R个弹珠中，一共有多少不同颜色的弹珠，“R x c”表示A把x位置上的弹珠换成了c颜色。

Output

对于每个Q操作，输出一行表示询问结果。

Sample Input

2 3
1 2
Q 1 2
R 1 2
Q 1 2

Sample Output

2
1

HINT

对于100%的数据，有1 ≤ N ≤ 10000, 1 ≤ M ≤ 10000，小朋友A不会修改超过1000次，所有颜色均用1到10^6的整数表示。

题解：

  这个跟教主的魔法差不多，问题转化过程贴hzwer

  http://hzwer.com/2793.html

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int MAXN = 1000001;
int pos[MAXN], a[MAXN], b[MAXN], add[MAXN];
int last[MAXN], pre[MAXN];
int n, m ,block;
void reset(int x)
{
    int i;
    int l = (x-1) * block + 1;
    int r = x * block;
    if (r > n) r = n;
    for (i = l; i <= r; i++)
    {
        pre[i] = b[i];
    }
    sort(pre + l, pre + r + 1);
}
int find(int x, int tar)
{
    int l = (x - 1)*block + 1;
    int r = x*block;
    if (r > n) r = n;
    int mid, ret=l;
    while (l <= r)
    {
        mid = (l + r) >> 1;
        if (pre[mid] >= tar) r = mid-1;
        else l = mid + 1;
    }
    return l-ret;
}
void Modify(int x, int tar)
{
    int i, temp;
    for (i = 1; i <= n; i++)
        last[a[i]] = 0;
    a[x] = tar;
    for (i = 1; i <= n; i++)
    {
        temp = b[i];
        b[i] = last[a[i]];
        if (temp != b[i]) reset(pos[i]);
        last[a[i]] = i;
    }
}
int Query(int L, int R)
{
    int d = 0, i;
    if (pos[L] == pos[R])
    {
        for (i = L; i <= R; i++)
        {
            if (b[i]<L)
                d++;
        }
    }
    else
    {
        for (i = L; i <= pos[L] * block; i++)
            if (b[i]<L) 
                d++;
        for (i = (pos[R]-1) * block + 1; i <= R; i++)
            if (b[i]<L) 
                d++;
    }
    for (i = pos[L] + 1; i < pos[R]; i++)
        d += find(i, L);
    return d;
}
int main(int argc, char *argv[])
{
    int i, j, x, y, z, T;
    char op;
    scanf("%d%d", &n, &T);
    block = int(sqrt(n) + log(2 * n) / log(2));
    //block = (int)sqrt(n);
    for (i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        b[i] = last[a[i]];
        last[a[i]] = i;
        pos[i] = (i - 1) / block + 1;
    }
    if (n%block)
        m = n / block + 1;
    else m = n / block;
    for (i = 1; i <= m; i++)
        reset(i);
    while (T--)
    {
        scanf("\n%c%d%d", &op, &x, &y);
        if (op == 'R') Modify(x, y);
        else printf("%d\n", Query(x, y));
    }
    return 0;
}
 
}