bzoj 2120 数颜色

Posted 2020-09-21

2120: 数颜色

Time Limit: 6 Sec  Memory Limit: 259 MB
Submit: 5039  Solved: 2012

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。

2016.3.2新加数据两组by Nano_Ape

 

莫队水之……

喜闻乐见的Code

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define maxn 10005
struct Query {
	int l,r,bl,br,num,cnt;
	bool operator < (const Query &a) const {
		if(bl!=a.bl) return bl<a.bl;
		else if(br!=a.br) return br<a.br;
		return cnt<a.cnt;
	}
	Query(int l=0,int r=0,int bl=0,int br=0,int num=0,int cnt=0):
		l(l),r(r),bl(bl),br(br),num(num),cnt(cnt) {}
}query[maxn];
struct Modify {
	int before,pos,after,num;
	Modify(int before=0,int pos=0,int after=0,int num=0):
		before(before),pos(pos),after(after),num(num) {}
}modify[maxn];
int qtime,mtime,col[maxn],color[maxn];
int numcolor,ans[maxn],cnt[maxn*1000]; 
inline void count(int pos,bool flag) {
	if(flag) {
		cnt[col[pos]]++;
		if(cnt[col[pos]]==1) numcolor++;
	} else {
		cnt[col[pos]]--;
		if(cnt[col[pos]]==0) numcolor--;
	}
	return;
}
int main() {
	int n,m,k,l,r;
	scanf("%d%d",&n,&m);
	k=sqrt(n+0.5);
	for(int i=1;i<=n;i++) {
		scanf("%d",&col[i]);
		color[i]=col[i];
	}
	for(int i=1;i<=m;i++) {
		char c[2];
		scanf("%s%d%d",c,&l,&r);
		if(c[0]==‘Q‘) query[qtime++]=Query(l,r,l/k,r/k,qtime,mtime);
		else modify[mtime++]=Modify(color[l],l,r,mtime),color[l]=r;
	}
	sort(query,query+qtime);
	l=1,r=0,m=0;
	for(int i=0;i<qtime;i++) {
		while(m<=query[i].cnt) {
			if(modify[m].pos>=l&&modify[m].pos<=r) {
				cnt[modify[m].before]--,cnt[modify[m].after]++;
				if(cnt[modify[m].before]==0) numcolor--;
				if(cnt[modify[m].after]==1) numcolor++;
			}
			col[modify[m].pos]=modify[m].after;
			m++;
		}
		while(m>query[i].cnt) {
			m--;
			if(modify[m].pos>=l&&modify[m].pos<=r) {
				cnt[modify[m].before]++,cnt[modify[m].after]--;
				if(cnt[modify[m].before]==1) numcolor++;
				if(cnt[modify[m].after]==0) numcolor--;
			}
			col[modify[m].pos]=modify[m].before;
		}
		while(l<query[i].l) count(l++,0);
		while(l>query[i].l) count(--l,1);
		while(r>query[i].r) count(r--,0);
		while(r<query[i].r) count(++r,1);
		ans[query[i].num]=numcolor;
	}
	for(int i=0;i<qtime;i++) printf("%d\n",ans[i]);
	return 0;
}