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

题目描述

题解

第一眼看过去是带修莫队。效率 $O(n^{\frac{5}{3}})$ 。可惜洛谷给出的标签是树套树，于是思考一下怎么用数据结构维护。

如果没有修改的话，我们考虑怎么在线计算答案，考虑到一种颜色只能计算一次，所以可以想到如果 $[l,r]$ 中有的颜色的前驱出现在 $[0,l-1]$ （没有即 $0$ ），那就造成贡献，因此我们可以用主席树维护 $[1,r]$ 中每个位置的前驱对应位置的个数。统计答案的时候就统计 $[l,r]$ 中 $[0,l-1]$ 的标记数即可。

考虑修改，那我们就用 $\text{set}$ 维护前驱，然后套树状数组即可。效率 $O(nlo{g}^{2}n)$ 。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+5,M=N*200;
int n,m,t,ls[M],T[N],rs[M],s[M],a[N];
set<int>st[1000005];
set<int>::iterator it;char ch[3];
#define mid ((l+r)>>1)
void upd(int& x,int l,int r,int p,int v){
	if (!x) x=++t;s[x]+=v;
	if (l==r) return;
	if (mid>=p) upd(ls[x],l,mid,p,v);
	else upd(rs[x],mid+1,r,p,v);
}
void upd(int x,int y,int v){
	for (;x<=n;x+=x&-x) upd(T[x],0,n,y,v);
}
int qry(int x,int l,int r,int R){
	if (!x) return 0;
	if (r<=R) return s[x];
	if (mid>=R) return qry(ls[x],l,mid,R);
	return s[ls[x]]+qry(rs[x],mid+1,r,R);
}
int qry(int l,int r,int p){
	int v=0;
	for (;r;r-=r&-r) v+=qry(T[r],0,n,p);
	for (;l;l-=l&-l) v-=qry(T[l],0,n,p);
	return v;
}
int main(){
	cin>>n>>m;
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i]),st[a[i]].insert(i);
	for (int i=1000000;i;i--) st[i].insert(0);
	for (int i=1;i<=n;i++)
		it=st[a[i]].lower_bound(i),it--,upd(i,*it,1);
	for (int u,v,l,r,i=1;i<=m;i++){
		scanf("%s%d%d",ch,&u,&v);
		if (ch[0]=='Q') printf("%d\n",qry(u-1,v,u-1));
		else{
			st[a[u]].erase(st[a[u]].find(u));
			it=st[a[u]].lower_bound(u);l=r=0;
			if (it!=st[a[u]].end()) r=*it;
			it--;l=*it;upd(u,l,-1);
			if (r) upd(r,u,-1),upd(r,l,1);
			it=st[v].lower_bound(u);l=r=0;
			if (it!=st[v].end()) r=*it;
			it--;l=*it;upd(u,l,1);
			if (r) upd(r,l,-1),upd(r,u,1);
			st[v].insert(u);a[u]=v;
		}
	}
	return 0;
}