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
1
HINT
对于100%的数据,有1 ≤ N ≤ 10000, 1 ≤ M ≤ 10000,小朋友A不会修改超过1000次,所有颜色均用1到10^6的整数表示。
Solve
ne[i]表示第i个位置,这种颜色的下一个在哪里,求区间内颜色个数,即为求区间内有多少个位置的下一个在区间外,同bzoj3343,块内二分,块外暴力
#include<algorithm>
#include<cstdio>
#include<cmath>
using namespace std;
const int N=10005;
int n,m,k,x,y,ans,a[N];
int be[N],l[N],r[N],len=1;
int ne[N],last[1000005],sor[N];
char ch[5];
inline void reset(int l,int r){
for (int i=l;i<=r;++i)sor[i]=ne[i];
sort(sor+l,sor+r+1);
}
int main (){
scanf ("%d%d",&n,&m);
k=sqrt(n);
for (int i=1;i<=n;++i){
scanf ("%d",&a[i]);
if (i%k==0)l[++len]=i;
r[(be[i]=len)]=i;
ne[i]=n+1;
ne[last[a[i]]]=i;
last[a[i]]=i;
}
for (int i=1;i<=len;++i)reset(l[i],r[i]);
for (int i=1;i<=m;++i){
scanf ("%s%d%d",ch,&x,&y);
if (ch[0]=='Q'){
ans=0;
if (be[x]==be[y]){
for (int j=x;j<=y;++j)if (ne[j]>y)ans++;
printf ("%d\n",ans);
}
else{
for (int j=x;j<=r[be[x]];++j)if (ne[j]>y)ans++;
for (int j=l[be[y]];j<=y;++j)if (ne[j]>y)ans++;
for (int j=be[x]+1;j<be[y];++j)
if (sor[r[j]]<=y)continue;
else ans+=r[j]-(upper_bound(sor+l[j],sor+r[j]+1,y)-sor)+1;
printf ("%d\n",ans);
}
}
else{
for (int j=1;j<=n;++j)last[a[j]]=n+1;
a[x]=y;last[a[x]]=n+1;
for (int j=n;j>=1;--j){
int t=ne[j];
ne[j]=last[a[j]];
if (t!=ne[j])reset(l[be[j]],r[be[j]]);
last[a[j]]=j;
}
}
}
return 0;
}