洛谷P1972&&BZOJ1878[SDOI2009]HH的项链

主席树的练手好题，中文题面不再赘述

对于这一道题，我们首先可以有一个很直观的做法：

对于询问区间[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;
}