主席树的练手好题,中文题面不再赘述
对于这一道题,我们首先可以有一个很直观的做法:
对于询问区间[i,j],直接用[i,j]减去重复颜色的数量。
那么我们如何计算重复颜色的数量呢?
对于第i个贝壳,我们设其颜色是col[i],重复就是在第i次之前也被计算了,我们在这里就仅仅考虑维护一个last[i]表示第i个贝壳的颜色上一次出现的位置,然后把第last[i]的位置在线段树中+1,设sum[i][j]=线段树中[1...i]对于第j个贝壳颜色last的累和,那么对于一段区间[l,r]中第i个贝壳重复出现的次数就是sum[i][r]-sum[i][l-1],其实这个差分是可以合并的,即重新令sum[i]=线段树中[1..i]中所有贝壳last的数量,那么这段区间里所有贝壳重复颜色的数量就是sum[r]-sum[l-1]。由于是对于任意区间,所以我们建n个线段树,再可持久化搞一搞就是好了。
代码
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=5e5+10;
const int M=N*10;
struct President_tree
{
int L,R,sum;
}t[M<<1];
int last[N],now[M],root[N];
int n,m,cnt=1;
void insert(int &rt,int x,int l,int r)
{
t[cnt++]=t[rt];rt=cnt-1;t[rt].sum++;
if (l==r)return;
int mid=(l+r)>>1;
if (x<=mid)insert(t[rt].L,x,l,mid);
else insert(t[rt].R,x,mid+1,r);
return ;
}
int query(int i,int j,int qr,int l,int r)
{
if (r<=qr)return t[j].sum-t[i].sum;
int mid=(l+r)>>1;
int ans=query(t[i].L,t[j].L,qr,l,mid);
if (mid<qr) ans+=query(t[i].R,t[j].R,qr,mid+1,r);
return ans;
}
int main()
{
scanf("%d",&n);int x;
for (int i=1;i<=n;i++)
{
scanf("%d",&x);
last[i]=now[x];now[x]=i;
}
root[0]=0;
for (int i=1;i<=n;i++)
root[i]=root[i-1],insert(root[i],last[i],0,n);
scanf("%d",&m);int l,r;
for (int i=1;i<=m;i++)
{
scanf("%d%d",&l,&r);
printf("%d\n",query(root[l-1],root[r],l-1,0,n));
}
return 0;
}