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
*/