洛谷 P1972 [SDOI2009]HH的项链

题目背景

无

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答……因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入输出格式

输入格式：

第一行：一个整数N，表示项链的长度。

第二行：N 个整数，表示依次表示项链中贝壳的编号（编号为0 到1000000 之间的整数）。

第三行：一个整数M，表示HH 询问的个数。

接下来M 行：每行两个整数，L 和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

输出格式：

M 行，每行一个整数，依次表示询问对应的答案。

输入输出样例

输入样例#1：

6
1 2 3 4 3 5
3
1 2
3 5
2 6

输出样例#1：

2
2
4

说明

数据范围：

对于100%的数据，N <= 50000，M <= 200000。

主席树

屠龙宝刀点击就送

#include <algorithm>
#include <cstdio>
#define N 50005
int tot,next[N],pre[N],rt[N],sum[N*100],ls[N*100],rs[N*100],n,m,a[N];
void update(int &y,int l,int r,int t)
{
    sum[++tot]=sum[y];
    ls[tot]=ls[y];
    rs[tot]=rs[y];
    ++sum[y=tot];
    if(l==r) return;
    int mid=(l+r)>>1;
    if(t<=mid) update(ls[y],l,mid,t);
    else update(rs[y],mid+1,r,t);
}
void getnext()
{
    for(int i=n;i>=1;--i) next[i]=pre[a[i]]?pre[a[i]]:n+1,pre[a[i]]=i; 
}
int query(int l,int r,int x,int y,int t)
{
    if(l==r) return t>l?sum[y]-sum[x]:0;
    int mid=(l+r)>>1;
    if(t<=mid) return query(l,mid,ls[x],ls[y],t)+sum[rs[y]]-sum[rs[x]];
    else return query(mid+1,r,rs[x],rs[y],t);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]);
    getnext();
    for(int i=1;i<=n;++i) rt[i]=rt[i-1],update(rt[i],1,n+1,next[i]);
    scanf("%d",&m);
    for(int l,r;m--;)
    {
        scanf("%d%d",&l,&r);
        printf("%d
",query(1,n+1,rt[l-1],rt[r],r));
    }
    return 0;
}