【BZOJ 2120】 数颜色

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。

正解是什么？我不知道

参考hzwer的思路，这题可以暴力分块，因为修改少。

记录每个颜色的上一次出现在那，然后排个序。边上暴力查找比上一个比l小的点，块内二分

话说lower_bound是查询前闭后开的区间。。。

 1 #include<cstdio>
 2 #include<cmath>
 3 #include<iostream>
 4 #include<algorithm>
 5 using namespace std;
 6 const int N=10010;
 7 int bel[N],last[1000001],size[N],pre[N],q[N],a[N],c[N];
 8 int num,n,m,l,r,p,col,ans;
 9
10 void build(){
11     int block=sqrt(n);
12     for(int i=1;i<=n;i++){
13         bel[i]=(i-1)/block+1;
14         if(bel[i]>num) num=bel[i];
15         size[bel[i]]++;
16         pre[i]=last[a[i]];
17         c[i]=pre[i];
18         last[a[i]]=i;
19     }
20 }
21
22 void reset(int x){
23     for(int i=q[x-1]+1;i<=q[x];i++) pre[i]=c[i];
24     sort(pre+q[x-1]+1,pre+q[x]+1);
25 }
26
27 int find(int x,int y){
28     int tmp=lower_bound(pre+q[x-1]+1,pre+q[x]+1,y)-pre-q[x-1]-1;
29     return tmp;
30 }
31
32 int ask(int l,int r){
33     if(bel[l]==bel[r]){
34         for(int i=l;i<=r;i++) if(c[i]<l) ans++;
35     }
36     else{
37         for(int i=l;i<=q[bel[l]];i++) if(c[i]<l) ans++;
38         for(int i=q[bel[r]-1]+1;i<=r;i++)
39         if(c[i]<l) ans++;
40     }
41     for(int i=bel[l]+1;i<bel[r];i++) ans+=find(i,l);
42     return ans;
43 }
44
45 void change(int x,int y){
46     for(int i=1;i<=n;i++) last[a[i]]=0;
47     a[x]=y;
48     for(int i=1;i<=n;i++){
49         int t=c[i];
50         c[i]=last[a[i]];
51         if(t!=c[i]) reset(bel[i]);
52         last[a[i]]=i;
53     }
54 }
55
56 int main(){
57     scanf("%d%d",&n,&m);
58     for(int i=1;i<=n;i++) scanf("%d",&a[i]);
59     build();
60     for(int i=1;i<=num;i++) q[i]=q[i-1]+size[i];
61     for(int i=1;i<=num;i++)
62     reset(i);
63     char s[4];
64     for(int i=1;i<=m;i++){
65         scanf("%s",s);
66         if(s[0]==‘Q‘){
67             ans=0;
68             scanf("%d%d",&l,&r);
69             printf("%d\n",ask(l,r));
70         }
71         else{
72             scanf("%d%d",&p,&col);
73             change(p,col);
74         }
75     }
76 }