BZOJ 2120: 数颜色 分块

2120: 数颜色

Time Limit: 20 Sec

Memory Limit: 256 MB

题目连接
http://www.lydsy.com/JudgeOnline/problem.php?id=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。

题意

两个操作，单点修改，以及查询区间有多少个数不一样

题解：

很显然，我们记录每个数他的前一个数的位置在哪儿，如果不存在就记录为-1

那么我们查询的时候，就只用查询[l,r]区间中小于l的数有多少个就好了，这个用平衡树解决就好了

下面是分块的代码，修改是o(n),查询是sqrt(n)*logn的，由于题目说了，修改操作很少，所以算是水过去了……

代码来自hzwer

代码：

//qscqesze
#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <typeinfo>
#include <fstream>
#include <map>
#include <stack>
typedef long long ll;
using namespace std;
//freopen("D.in","r",stdin);
//freopen("D.out","w",stdout);
#define sspeed ios_base::sync_with_stdio(0);cin.tie(0)
#define maxn 2000051
#define mod 10007
#define eps 1e-9
int Num;
//const int inf=0x7fffffff;   //нчоч╢С
const int inf=0x3f3f3f3f;
inline ll read()
{
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
//**************************************************************************************

int n,q,m,block;
int c[maxn],pos[maxn],pre[maxn],b[maxn],last[maxn];

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 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;
}
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()
{
n=read(),q=read();
for(int i=1;i<=n;i++)
c[i]=read();
block = int(sqrt(n));
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;
}