[国家集训队] 数颜色
题目链接:戳我
带修莫队。时间复杂度\(O(N^{5/3})\)
不会的可以看看胡小兔 dalao的博客
先离线所有操作,然后标记上修改的时间戳,还按照普通莫队做,如果当前处理的询问在上一个处理的时间之前,就还原之间所有操作,如果在之后就先进行之间的所有操作。
排序按照块编号为第一关键字,右端点为第二关键字。
代码如下:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define MAXN 1000010
using namespace std;
int n,m,cnt1,cnt2,ans,len;
int c[MAXN],cnt[MAXN],a[MAXN],ansans[MAXN];
struct Node{int l,r,id,t,ti;}q[MAXN];
struct Node2{int x,c,last;}node[MAXN];
inline bool cmp(struct Node x,struct Node y)
{
if(x.t!=y.t) return x.t<y.t;
return x.r<y.r;
}
inline void change(int pre,int now,int l,int r)
{
if(pre==now) return;
if(pre<now)
{
for(int i=pre+1;i<=now;i++)
{
if(node[i].x<l||node[i].x>r) c[node[i].x]=node[i].c;
else
{
cnt[c[node[i].x]]--;
if(cnt[c[node[i].x]]==0) ans--;
cnt[node[i].c]++;
if(cnt[node[i].c]==1) ans++;
c[node[i].x]=node[i].c;
}
}
}
else
{
for(int i=pre;i>now;i--)
{
if(node[i].x<l||node[i].x>r) c[node[i].x]=node[i].last;
else
{
cnt[c[node[i].x]]--;
if(cnt[c[node[i].x]]==0) ans--;
cnt[node[i].last]++;
if(cnt[node[i].last]==1) ans++;
c[node[i].x]=node[i].last;
}
}
}
}
inline void count(int x,int k)
{
cnt[c[x]]+=k;
if(cnt[c[x]]==0&&k==-1) ans--;
if(cnt[c[x]]==1&&k==1) ans++;
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("ce.in","r",stdin);
#endif
scanf("%d%d",&n,&m);
len=sqrt(n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),c[i]=a[i];
for(int i=1;i<=m;i++)
{
char ch;int x,y;
cin>>ch;
scanf("%d%d",&x,&y);
if(ch=='Q')
cnt1++,q[cnt1]=(Node){x,y,cnt1,(x-1)/len+1,cnt2};
else
{
cnt2++,node[cnt2]=(Node2){x,y,c[x]};
c[x]=y;
}
}
for(int i=1;i<=n;i++) c[i]=a[i];
sort(&q[1],&q[cnt1+1],cmp);
int l=1,r=0,pre=0;
for(int i=1;i<=cnt1;i++)
{
if(pre!=q[i].ti) change(pre,q[i].ti,l,r);
while(l<q[i].l) count(l++,-1);
while(l>q[i].l) count(--l,+1);
while(r<q[i].r) count(++r,+1);
while(r>q[i].r) count(r--,-1);
ansans[q[i].id]=ans;
pre=q[i].ti;
}
for(int i=1;i<=cnt1;i++)
printf("%d\n",ansans[i]);
return 0;
}