bzoj 2453（分块）

2453: 维护队列

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 558   Solved: 240
[ Submit][ Status][ Discuss]

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,2,3,4。修改就改变值。用pre[i]表示与当前不相同的最近的点。对pre数组在每个快中排序，然后就只要二分查找就可以了。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define N 10001
#define M 1000001
using namespace std;
int n,q,m,block;
int c[N],pos[N],pre[N],b[N],last[M];
int find(int x,int v)
{
int l=(x-1)*block+1,r=min(x*block,n);
int first=l;
while(l<=r)
{
int mid=(l+r)>>1;
if(pre[mid]<v)l=mid+1;
else r=mid-1;
}
return l-first;
}
void reset(int x)
{
int l=(x-1)*block+1,r=min(x*block,n);
for(int i=l;i<=r;i++)pre[i]=b[i];
sort(pre+l,pre+r+1);
}
void build()
{
for(int i=1;i<=n;i++)
{
b[i]=last[c[i]];
last[c[i]]=i;
pos[i]=(i-1)/block+1;
}
for(int i=1;i<=m;i++)reset(i);
}
int ask(int l,int r)
{
int ans=0;
if(pos[l]==pos[r])
{
for(int i=l;i<=r;i++)if(b[i]<l)ans++;
}
else 
{
for(int i=l;i<=block*pos[l];i++)if(b[i]<l)ans++;
for(int i=block*(pos[r]-1)+1;i<=r;i++)if(b[i]<l)ans++;
}
for(int i=pos[l]+1;i<pos[r];i++)
   ans+=find(i,l);
return ans;
}
void change(int x,int v)
{
for(int i=1;i<=n;i++)last[c[i]]=0;
c[x]=v;
for(int i=1;i<=n;i++)
{
int t=b[i];
b[i]=last[c[i]];
if(t!=b[i])reset(pos[i]);
last[c[i]]=i;
}
}
int main()
{
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)scanf("%d",&c[i]);
block=int(sqrt(n)+log(2*n)/log(2));
if(n%block)m=n/block+1;
else m=n/block;
build();
char ch[5];int x,y;
for(int i=1;i<=q;i++)
{
scanf("%s%d%d",ch,&x,&y);
if(ch[0]=='Q')printf("%d\n",ask(x,y));
else change(x,y);
}
return 0;
}