【Bzoj1878】HH的项链

1878: [SDOI2009]HH的项链

Time Limit: 4 Sec   Memory Limit: 64 MB
Submit: 3770   Solved: 1874
[ Submit][ Status][ Discuss]

Description

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

Input

第一行：一个整数N，表示项链的长度。 第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。 第三行：一个整数M，表示HH询问的个数。 接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

Output

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

Sample Input

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

Sample Output

2
2
4

HINT

对于20%的数据，N ≤ 100，M ≤ 1000；
对于40%的数据，N ≤ 3000，M ≤ 200000；
对于100%的数据，N ≤ 50000，M ≤ 200000。

这题目很多人看第一眼会想到莫队，这的确是一个莫队的好题，但是这里介绍一种方法，直接树状数组+离线处理就能搞。首先读入时，记录每种颜色的第一次出现位置，在树状数组中加上，以及每一个贝壳下一个同色贝壳的位置，之后我们将询问读入，然后按左端点排序。之后考虑询问，放一个标记一直往右加，也就是说将每次左端点前的贝壳的下一个同色贝壳激活（因为左边的那些贝壳之后再也不会被询问到。）然后树状数组求区间和就行了。

#include <cstdio>
#include <algorithm>

using namespace std;

const int maxx = 50000 + 100;
const int maxn = 1000000 + 100;
const int maxm = 200000 + 100;

int n,m,tot;
int a[maxx],t[maxx],next[maxx];
int head[maxn];

struct Que{
    int l;
    int r;
    int id;
    int ans;
}Q[maxm];

inline int Read(){
    int x=0,f=1;char c=getchar();
    while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') {x=x*10+c-'0'; c=getchar();}
    return x*f;
}

int lowbit(int x){
    return x&(-x);
}

void Add(int x,int k){
    for(int i=x;i<=n;i+=lowbit(i))
        t[i] += k;
}

int Query(int x){
    int tmp = 0;
    for(int i=x;i>=1;i-=lowbit(i))
        tmp += t[i];
    return tmp;
}

bool cmp1(Que a,Que b){
    return a.l==b.l? a.r<b.r : a.l<b.l;
}

bool cmp2(Que a,Que b){
    return a.id<b.id;
}

int main(){
    n = Read();
    for(int i=1;i<=n;i++)
        a[i]=Read(),a[i]++,tot=max(tot,a[i]);
    for(int i=n;i>=1;i--)
        next[i]=head[a[i]],head[a[i]]=i;
    for(int i=1;i<=tot;i++)
        if(head[i]) Add(head[i],1);
    m = Read();
    for(int i=1;i<=m;i++)
        Q[i].l=Read(),Q[i].r=Read(),Q[i].id=i;
    sort(Q+1,Q+m+1,cmp1);
    int now=1;
    for(int i=1;i<=m;i++){
        while(now<Q[i].l){
            if(next[now])
                Add(next[now],1);
            now++;
        }
        Q[i].ans=Query(Q[i].r)-Query(Q[i].l-1);
    }
    sort(Q+1,Q+m+1,cmp2);
    for(int i=1;i<=m;i++)
        printf("%d\n",Q[i].ans);
    return 0;
}