BZOJ2120 数颜色 莫队 带修莫队

原文链接https://www.cnblogs.com/zhouzhendong/p/BZOJ2120.html

题目传送门 - BZOJ2120

题意

  给定一个长度为 $n$ 的序列 $a$ ,有 $m$ 次操作,每次操作可能是如下两种类型之一:

  1. 询问区间不同值的个数。

  2. 修改一个 $a_i$ 。

  对于每一次询问,输出结果。

  $n,m\leq 10000,\ \ \ \ 1\leq a_i\leq10^6$

题解

  带修莫队裸题。

  打错一个变量找了20分钟。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=10005,M=1000005;
int n,m,Qcnt=0,Ocnt=0;
int c[N],dc[N],k[N];
int tax[M],tot=0;
struct Query{
int L,R,id,ans;
Query(){}
Query(int _L,int _R,int _id){
L=_L,R=_R,id=_id;
}
}Q[N];
struct Operation{
int x,v1,v2,id;
Operation(){}
Operation(int _x,int _v1,int _v2,int _id){
x=_x,v1=_v1,v2=_v2,id=_id;
}
}O[N];
bool cmp(Query a,Query b){
if (k[a.L]!=k[b.L])
return k[a.L]<k[b.L];
if (k[a.R]!=k[b.R])
return k[a.R]<k[b.R];
return a.id<b.id;
}
bool cmpid(Query a,Query b){
return a.id<b.id;
}
int L=1,R=0,t=0;
void update(int t){
int x=O[t].v1,y=O[t].v2,id=O[t].x;
if (c[id]!=x)
swap(x,y);
if (L<=id&&id<=R){
tot-=(--tax[x])==0;
tot+=(tax[y]++)==0;
}
c[id]=y;
}
int main(){
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",&c[i]),dc[i]=c[i];
for (int i=1;i<=m;i++){
char ch[2];
int x,y;
scanf("%s%d%d",ch,&x,&y);
if (ch[0]=='Q')
Q[++Qcnt]=Query(min(x,y),max(x,y),i);
else {
O[++Ocnt]=Operation(x,y,dc[x],i);
dc[x]=y;
}
}
for (int i=1;i<=n;i++)
k[i]=(i-1)/400;
sort(Q+1,Q+Qcnt+1,cmp);
memset(tax,0,sizeof tax);
O[0].id=0,O[Ocnt+1].id=m+1;
for (int i=1;i<=Qcnt;i++){
while (L<Q[i].L)
tot-=(--tax[c[L++]])==0;
while (L>Q[i].L)
tot+=(tax[c[--L]]++)==0;
while (R<Q[i].R)
tot+=(tax[c[++R]]++)==0;
while (R>Q[i].R)
tot-=(--tax[c[R--]])==0;
while (O[t+1].id<Q[i].id)
update(++t);
while (O[t].id>Q[i].id)
update(t--);
Q[i].ans=tot;
}
sort(Q+1,Q+Qcnt+1,cmpid);
for (int i=1;i<=Qcnt;i++)
printf("%d\n",Q[i].ans);
return 0;
}

  

上一篇:Auto X2021 K Increasing Sequence


下一篇:[题解]P1280