BZOJ2120 - 数颜色

http://www.lydsy.com/JudgeOnline/problem.php?id=2120

莫队的主要思想，就是尽量控制区间转移的次数。

本题与一般莫队的区别在于加入了修改操作。然而事实上，我们可以把修改看成查询的另一个参数，即查询由Q(ft,to)变为了Q(fr,to,mmt)，代表查询mmt时刻（每进行一次修改时刻就+1）时的区间[fr,to]。那么只需找出新的转移方法即可。

fr与to的转移很简单，在此略过。

重点说一下mmt的转移。当我们存储修改操作R时，可以额外存储一个变量col0，记录该次修改前p的颜色，表示这次修改将p的颜色从col0改成了col。那么当mmt回退时，就把p改成col0，反之当mmt前进时，把p改成col。

//数颜色 - 带修改莫队 
#include <cstdio>
#include <algorithm>
using namespace std;
int const N=10010;
int n,m,a[N];
int fr0,to0,mmt0,tAns;
struct query
{
	int fr,to,mmt;
	int ori;
}q[N];
int fCol[N];
struct repaint
{
	int p,col,col0;
}r[N];
int n0,cnt[N*100],ans[N];
bool rngCmp(query q1,query q2)
{
	int rng1=(q1.fr-1)/n0,rng2=(q2.fr-1)/n0;
	if(rng1==rng2)
		if(q1.to==q2.to) return q1.mmt<q2.mmt;
		else return q1.to<q2.to;
	else return rng1<rng2;	
}
void paint(int p,int col)
{
	if(fr0<=p && p<=to0)
	{
		if(cnt[a[p]]--==1) tAns--;
		if(++cnt[col]==1) tAns++;
	}
	a[p]=col;
}
int main()
{
	freopen("bzoj2120.in","r",stdin);
	//读入
	scanf("%d%d",&n,&m);
	while(n0*n0<n) n0++;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),fCol[i]=a[i];
	int k=0,mQ=0,mR=0;
	for(int i=1;i<=m;i++) 
	{
		char ch;
		scanf("\n%c",&ch);
		if(ch=='Q')
		{
			mQ++;
			scanf("%d%d",&q[mQ].fr,&q[mQ].to);
			q[mQ].mmt=k;
			q[mQ].ori=i;
		}
		else
		{
			mR++;
			scanf("%d%d",&r[mR].p,&r[mR].col);
			int p=r[mR].p,col=r[mR].col;
			r[mR].col0=fCol[p];
			fCol[p]=col;
			k++;
		}
	}
	//莫队算法
	sort(q+1,q+mQ+1,rngCmp);
	/*for(int i=1;i<=mQ;i++) printf("[%d,%d] ",q[i].fr,q[i].to);
	printf("\n");
	for(int i=1;i<=mR;i++) printf("%d,%d->%d ",r[i].p,r[i].col0,r[i].col);
	printf("\n");*/
	//求解
	fr0=1,to0=0,mmt0=0;
	tAns=0;
	for(int i=1;i<=mQ;i++)
	{
		int fr=q[i].fr,to=q[i].to,mmt=q[i].mmt;
		while(fr<fr0) if(++cnt[a[--fr0]]==1) tAns++;
		while(fr>fr0) if(cnt[a[fr0++]]--==1) tAns--;
		while(to<to0) if(cnt[a[to0--]]--==1) tAns--;
		while(to>to0) if(++cnt[a[++to0]]==1) tAns++;
		while(mmt<mmt0)
		{
			paint(r[mmt0].p,r[mmt0].col0);
			mmt0--;
		}
		while(mmt>mmt0)
		{
			mmt0++;
			paint(r[mmt0].p,r[mmt0].col);
		}
		/*for(int j=1;j<=6;j++) printf("%d ",cnt[j]);
		printf("\n");*/
		ans[q[i].ori]=tAns;
	}
	for(int i=1;i<=m;i++) if(ans[i]) printf("%d\n",ans[i]);
	return 0;
}

P.S. 这道题因为数据范围太小或者数据上的原因，暴力也能过…