【bzoj2120】数颜色 莫队

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2120

【题解】

一道很简单的模板题——带修改的莫队

将修改操作与查询操作分开，对于每个查询操作，我们记录一下离他最近的修改操作的编号。

在进行转移的时候不仅需要维护区间，更需要去维护时间。

就是对于当前询问，在这之前的没有修改的修改操作要进行修改，而这之后的已经修改的修改操作要恢复去。

其他的就与普通的莫队算法一样了。

/*************
  bzoj 2120
  by chty
  2016.11.23
*************/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
#define FILE "read"
#define MAXN 10010
#define up(i,j,n) for(ll i=j;i<=n;i++)
#define down(i,j,n) for(ll i=j;i>=n;i--)
struct node{ll x,y,id,time;}q[MAXN];
struct node2{ll pos,col,pre;}w[MAXN];
ll n,m,cnt1,cnt2,block,now,a[MAXN],last[MAXN],ans[MAXN],vis[MAXN],num[1000100];
char ch[5];
namespace Init{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline ll read(){
		ll x=0,f=1;  char ch=getchar();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getchar();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getchar();}
		return x*f;
	}
}using namespace Init;
namespace Mo_Team{
	bool cmp(node a,node b) {return a.x/block==b.x/block?a.y<b.y:a.x/block<b.x/block;}
	void cal(ll x){
		if(vis[x]&&--num[a[x]]==0) now--;
		if(!vis[x]&&++num[a[x]]==1)  now++;
		vis[x]^=1;
	}
	void change(ll x,ll y){
		if(vis[x]) {cal(x); a[x]=y; cal(x);}
		else a[x]=y;
	}
	void work(){
		block=(ll)sqrt(n*1.0);
		sort(q+1,q+cnt2+1,cmp);
		ll l=1,r=0;
		up(i,1,cnt2){
			up(j,q[i-1].time+1,q[i].time)   change(w[j].pos,w[j].col);
			down(j,q[i-1].time,q[i].time+1) change(w[j].pos,w[j].pre);
			while(q[i].y>r)  cal(++r);
			while(q[i].y<r)  cal(r--);
			while(q[i].x>l)  cal(l++);
			while(q[i].x<l)  cal(--l);
			ans[q[i].id]=now;
		}
		up(i,1,cnt2)  printf("%lld\n",ans[i]);
	}
}
void init(){
	n=read();  m=read();
	up(i,1,n)  a[i]=read(),last[i]=a[i];
	up(i,1,m){
		scanf("%s",ch+1);  ll x=read(),y=read();
		if(ch[1]=='R'){w[++cnt1].pos=x; w[cnt1].col=y; w[cnt1].pre=last[x]; last[x]=y;}
		else {q[++cnt2].x=x; q[cnt2].y=y; q[cnt2].id=cnt2; q[cnt2].time=cnt1;}
	}
}
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	init();
	Mo_Team::work();
	return 0;
}