2120. [国家集训队]数颜色莫队

Posted 2020-10-29 Refun

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。

 

莫队模板题

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define MAXN (1000000+10)
using namespace std;
struct node
{
    int l,r,ord,id,ans,time;
}Ask[MAXN];
struct node1
{
	int val,pre,pos;
}Update[MAXN];
int l=1,r=0,now=0,n,m,Color_sum;
int cnt[MAXN],a[MAXN],Anum,Unum; 
bool cmp1(node a,node b){return a.id==b.id?(a.r==b.r?a.time<b.time:a.r<b.r):a.id<b.id;}
bool cmp2(node a,node b){return a.ord<b.ord;}

void Del(int x)
{
    cnt[x]--;
    if (!cnt[x]) Color_sum--;
}
void Ins(int x)
{
    if (!cnt[x]) Color_sum++;
    cnt[x]++;
}
void Recovery(int x,int val)
{
	if (l<=x && x<=r)
	{
		Ins(val);
		Del(a[x]);
	}
	a[x]=val;
}
void MoQueue(int x)
{
    int L=Ask[x].l,R=Ask[x].r;
    while (now<Ask[x].time) Recovery(Update[now+1].pos,Update[now+1].val),now++;
    while (now>Ask[x].time) Recovery(Update[now].pos,Update[now].pre),now--;
	while (l>L) Ins(a[--l]);
    while (l<L) Del(a[l++]);
    while (r>R) Del(a[r--]);
    while (r<R) Ins(a[++r]);
    Ask[x].ans=Color_sum;
}

int main()
{
	int x,y;
	char p;
	scanf("%d%d",&n,&m);
	int len=pow(n,2.0/3.0);
	for (int i=1;i<=n;++i)
		scanf("%d",&a[i]);
	for (int i=1;i<=m;++i)
	{
		scanf("\n%c%d%d",&p,&x,&y);
		if (p==‘Q‘)
		{
			Ask[++Anum].l=x;Ask[Anum].r=y;
			Ask[Anum].id=Ask[Anum].l/len;
			Ask[Anum].ord=Anum;
			Ask[Anum].time=Unum;
		}
		if (p==‘R‘)
		{
			Update[++Unum].pos=x;
			Update[Unum].val=y;	
			Update[Unum].pre=a[x];
			a[x]=y;
		}
	}
	for (int i=Unum;i>=1;--i)//恢复a区间 
		a[Update[i].pos]=Update[i].pre;	
	sort(Ask+1,Ask+Anum+1,cmp1);
	for (int i=1;i<=m;++i)
		MoQueue(i);
	sort(Ask+1,Ask+Anum+1,cmp2);
	for (int i=1;i<=Anum;++i)
		printf("%d\n",Ask[i].ans);
}