【BZOJ】1878: [SDOI2009]HH的项链 （主席树）

题目

传送门：QWQ

---

分析

莫队也能做，但我想练练主席树。

求k-th一样维护第i个时候的线段树，线段树来维护区间不同数。

然后查询时可以通过上下界小优化一波。

但是我的代码丑陋无比，常数巨大（捂脸

---

代码

#include <bits/stdc++.h>
using namespace std;

const int maxn=;
int ls[maxn<<], rs[maxn<<], sum[maxn<<], newp, root[maxn<<];
int last[];

inline int in()
{
    int x=;char ch=getchar();
    while(ch<''||ch>''){ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x;
}

void add(int l,int r,int x,int& cur,int cur1,int val)
{
    cur=++newp;
    ls[cur]=ls[cur1]; rs[cur]=rs[cur1];// root[i]=root[i-1]
    sum[cur]=sum[cur1]+val;
    if(l==r) return;
    int mid=l+r>>;
    if(x<=mid) add(l,mid,x,ls[cur],ls[cur1],val);
    else add(mid+,r,x,rs[cur],rs[cur1],val);
}

int query(int l,int r,int cnt,int cnt1,int L)
{
    if(l>=L) return sum[cnt1]-sum[cnt];
    int mid=l+r>>,ans=;
    if(mid>=L) ans+=query(l,mid,ls[cnt],ls[cnt1],L);
     ans+=query(mid+,r,rs[cnt],rs[cnt1],L);
    return ans;
}

int main()
{
    int n=in();
    for(int i=;i<=n;i++)
    {
        int x,temp; x=in();
        if(last[x])
        {
            add(,n,last[x],root[i],root[i-],-);
            add(,n,i,root[i],root[i],);
        }
        else
        {
            add(,n,i,root[i],root[i-],);
        }
        last[x]=i;
    }
    int q;
    q=in();
    for(int i=;i<=q;i++)
    {
        int l,r;
        l=in(); r=in();
        printf("%d\n",query(,n,root[l-],root[r],l));
    }
}