BZOJ2120［数颜色］

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。

---

solution: 分块＋二分

PS:与bzoj2453一模一样

#include <cmath>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

#define M 1000005
#define N 10005

int a[N], b[N], pre[N], bl[N], last[M];
int blk, cnt, n, m;

int cmp( int a, int b){
    return a < b;
}

void reset( int x ){
    int l = (x-1)*blk+1, r = min( x*blk, n);
    for ( int i = l; i <= r; i++) pre[i] = b[i];
    sort( pre+l, pre+r+1, cmp);
}

void build(){
    for ( int i = 1; i <= n; i++){
        b[i] = last[ a[i] ];
        last[ a[i] ] = i;
        bl[i] = (i-1)/blk + 1;
    }
    for ( int i = 1; i <= cnt; i++) reset( i );
}


int find( int x, int v ){
    int l = ( x-1 )*blk + 1, r = min(x * blk, n);
    int last = l;
    while ( l <= r ){
        int mid = ( l + r ) >> 1;
        if ( pre[mid] < v ) l = mid+1;
        else r = mid - 1;
    }
    return l - last;
}

int ask( int x, int y ){
    int ans = 0;
    if ( bl[x] == bl[y] ){
        for ( int i = x; i <= y; i++) if ( b[i] < x ) ans ++;
    }else {
        for ( int i = x; i <= bl[x]*blk; i++) if ( b[i] < x) ans ++;
        for ( int i = (bl[y]-1)*blk+1 ; i <= y; i++) if ( b[i] < x ) ans ++;
    }
    for ( int i = bl[x]+1; i <= bl[y]-1; i++) ans += find( i, x );
    return ans;
}

void change( int x, int v ){
    for(int i=1;i<=n;i++) last[a[i]]=0;
    a[x] = v;
    for ( int i = 1; i <= n; i++){
        int t = b[i];
        b[i] = last[ a[i] ];
        if ( t != b[i] ) reset( bl[i] );
        last[ a[i] ] = i;
    }

}


int main(){
    scanf( "%d%d", &n, &m );
    for ( int i = 1; i <= n; i++) scanf( "%d", &a[i] );
    blk = (sqrt(n)+log(2*n)/log(2));
    if ( n % blk ) cnt = n / blk ;
    else cnt = n / blk + 1;
    build();
    char ch[5]; int x, y;
    while ( m-- ){
        scanf( "%s%d%d", ch, &x, &y );
        if ( ch[0] == 'Q' ) printf( "%d\n", ask( x, y ) );
        else change( x, y);
    }
    return 0;
}