BZOJ 2120: 数颜色&&2453: 维护队列【双倍经验】

分块大法好啊

（不会主席树也不会莫队我好菜啊）

用前驱数组记录位置i之前的第一个与它颜色相同的位置pre[i]

因为修改次数少，可以直接暴力搞

询问的话显然如果pre[i]不在[x,y]中就是新的颜色(++ans),所以对于整块可以以pre为关键字排序然后二分找即可

(两道题数据范围貌似都一样啊)

#include<cstdio>
#include<algorithm>
#include<cmath>
#define mid ((l+r)>>1)
#define N 10005
using namespace std;

int read()
{
	int a=0,f=1;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}
	return a*f;
}

int qn,n,m,size;
int b[N],pre[N],col[N],l[110],r[110],bl[N],last[1000005];

void rebuild(int x)
{
	for(int i=l[x];i<=r[x];++i) b[i]=pre[i];
	sort(b+l[x],b+r[x]+1);
}

int solve(int x,int y,int z)
{
	int l=x,r=y,ans=x-1;
	while(l<=r)
	{
		if(b[mid]>=z) r=mid-1;
		else ans=mid,l=mid+1;
	}
	return ans-x+1;
}

int ask(int x,int y)
{
	int ans=0;
	if(bl[x]==bl[y])
		for(int i=x;i<=y;++i){if(pre[i]<x)++ans;}
	else
	{
		for(int i=bl[x]+1;i<=bl[y]-1;++i) ans+=solve(l[i],r[i],x);
		for(int i=x;i<=r[bl[x]];++i) if(pre[i]<x) ++ans;
		for(int i=l[bl[y]];i<=y;++i) if(pre[i]<x) ++ans;
	}
	return ans;
}

void change(int x,int y)
{
	for(int i=1;i<=qn;++i) last[col[i]]=0;
	col[x]=y;
	for(int i=1;i<=qn;++i)
	{
		int tmp=pre[i];
		pre[i]=last[col[i]];
		if(pre[i]!=tmp) rebuild(bl[i]);
		last[col[i]]=i;
	}
}

int main()
{
	qn=read(),m=read();
	size=(int)sqrt(qn),n=(qn-1)/size+1;
	for(int i=1;i<=qn;++i)
	{
		col[i]=read();
		b[i]=pre[i]=last[col[i]];
		last[col[i]]=i;
		bl[i]=(i-1)/size+1;
		if(!l[(i-1)/size+1]) l[(i-1)/size+1]=i;
		r[(i-1)/size+1]=i;
	}
	for(int i=1;i<=n;++i) sort(b+l[i],b+r[i]+1); 
	while(m--)
	{
		char c=getchar();
		while(c!='Q'&&c!='R') c=getchar();
		int x=read(),y=read();
		if(c=='Q')	printf("%d\n",ask(x,y));
		else change(x,y);
	}
	return 0;
}