P1903 [国家集训队]数颜色 / 维护队列

最新推荐文章于 2025-05-27 19:34:15 发布

原创最新推荐文章于 2025-05-27 19:34:15 发布 · 132 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法-莫队

算法-莫队专栏收录该内容

0 篇文章

订阅专栏

本文介绍了一种解决国家集训队数颜色问题的方法，使用带修莫队算法优化处理队列修改操作，实现对区间内不同颜色数量的快速查询。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

P1903 [国家集训队]数颜色 / 维护队列

题目描述

详见：P1903 [国家集训队]数颜色 / 维护队列

solution

裸的带修莫队。

多记录一维询问时已经进行多少次修改。

用莫队思想暴力即可。

（不开O2时需要卡常，洛谷评测两发一样的代码，一个TLE4个点，一个AC，QAQ）

code

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e6+50;
int a[MAXN],cnt[MAXN],Ans[MAXN];
struct pnode{ int id,l,r,x1,x2,t; } p[MAXN];
struct qnode{ int x,pre,nxt; } q[MAXN];
int compare(pnode x,pnode y){ return (x.x1<y.x1)||(x.x1==y.x1&&x.x2<y.x2)||(x.x1==y.x1&&x.x2==y.x2&&x.t<y.t); }
inline int read()
{
	int f=1,x=0; char c=getchar();
	while (c<'0'||c>'9') { if (c=='-') f=-1; c=getchar(); }
	while (c>='0'&&c<='9') { x=(x<<3)+(x<<1)+(c^48); c=getchar(); }
	return x*f;
}
inline char get_ch()
{
	char c=getchar();
	while (c<'A'||c>'Z') c=getchar();
	return c;
}
int main()
{
	int n=read(),m=read(),numq=0,nump=0;
	for (int i=1;i<=n;i++) a[i]=read();
	for (int i=1;i<=m;i++)
	{
		char c=get_ch(); 
		int x=read(),y=read();
		if (c=='Q') p[++nump]=(pnode){nump,x,y,0,0,numq};	
		else
		{
			q[++numq]=(qnode){x,a[x],y};
			a[x]=y;
		}
	}
	int Size=ceil(exp((log(n)+log(numq))/3));
	for (int i=1;i<=nump;i++) p[i].x1=(p[i].l-1)/Size+1,p[i].x2=(p[i].r-1)/Size+1;
	sort(p+1,p+nump+1,compare);
	
	int L=1,R=0,T=numq,num=0;
	memset(cnt,0,sizeof cnt);
	for (int i=1;i<=nump;i++)
	{
		while (L<p[i].l) cnt[a[L]]--,num-=(cnt[a[L]]==0),L++;
		while (L>p[i].l) L--,cnt[a[L]]++,num+=(cnt[a[L]]==1);
		while (R<p[i].r) R++,cnt[a[R]]++,num+=(cnt[a[R]]==1);
		while (R>p[i].r) cnt[a[R]]--,num-=(cnt[a[R]]==0),R--;
		
		while (T>p[i].t) 
		{
			qnode Q=q[T];
			if (p[i].l<=Q.x&&Q.x<=p[i].r)
			{
				cnt[Q.nxt]--; num-=(cnt[Q.nxt]==0);
				cnt[Q.pre]++; num+=(cnt[Q.pre]==1);
			}
			a[Q.x]=Q.pre;
			T--;
		}
		while (T<p[i].t) 
		{
			T++;
			qnode Q=q[T];
			if (p[i].l<=Q.x&&Q.x<=p[i].r)
			{
				cnt[Q.pre]--; num-=(cnt[Q.pre]==0);
				cnt[Q.nxt]++; num+=(cnt[Q.nxt]==1);
			}
			a[Q.x]=Q.nxt;
		}
		Ans[p[i].id]=num;
	}
	for (int i=1;i<=nump;i++) printf("%d\n",Ans[i]);
	return 0;
} 
/*
6 5
1 2 3 4 5 5
Q 1 4
Q 2 6
R 1 2
Q 1 4
Q 2 6

4
4
3
4
*/