洛谷1972 [SDOI2009]HH的项链

题目描述

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

输入格式：

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

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

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

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

输出格式：

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

 

可以树状数组求前缀和来离线做，在线做似乎需要主席树，菜鸡的我并不会主席树。。

首先用p数组记录一下每种颜色最左端的位置，用next数组来记录每种颜色下一个的位置

把询问按照左端点排序，碰到一个点x就将a[next[x]]++

每次询问就是sum[r]-sum[l-1] 

sum数组是a的前缀和

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
#define maxn 50005
#define maxm 200005
int a[maxn],c[maxn];
struct data{
    int l,r,ans,id;
}q[maxm];
int next[maxn],p[maxn*20];
int n,m;
int lowbit(int x){return x&(-x);}
bool cmp1(data a,data b){
    return a.l==b.r?a.r<b.r : a.l<b.l;
}
bool cmp2(data a,data b){
    return a.id<b.id;
}
void update(int x,int v){
    for(int i=x;i<=n;i+=lowbit(i)){
        c[i]+=v;
    }
}
int ask(int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i))res+=c[i];
    return res;
}
int mx;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        mx=max(mx,a[i]);
    }
    for(int i=n;i>0;i--){
        next[i]=p[a[i]];
        p[a[i]]=i;
    }
    for(int i=0;i<=mx;i++){
        if(p[i])update(p[i],1);
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    sort(q+1,q+1+m,cmp1);
    int l=1;
    for(int i=1;i<=m;i++){
        while(l<q[i].l){
            if(next[l])update(next[l],1);
            l++;
        }
        q[i].ans=ask(q[i].r)-ask(q[i].l-1);
    }
    sort(q+1,q+1+m,cmp2);
    for(int i=1;i<=m;i++){
        printf("%d\n",q[i].ans);
    }
}