[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 <= 500000，M <= 500000。

 

树状数组+离线

先按r排序，这样就可以转化为区间和问题

如果后面数与前面的有一样的就将前面的-1，当前位+1

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

const int maxn = 500010;

const int maxm = 1e6+10;

int n,a[maxn],m,tree[maxn],las[maxm],ans[maxn];

struct node {
    int l,r,id;
}qj[maxn];

bool cmp(const node& a,const node& b) {
    return a.r<b.r;
}

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

void add(int x,int sum) {
    while(x <= n) {
        tree[x] += sum;
        x += lowbit(x);
    }
} 

int getsum(int x) {
    int sum = 0;
    while(x) {
        sum += tree[x];
        x -= lowbit(x);
    }
    return sum;
}

int main() {
    scanf("%d",&n);
    for(int i = 1;i <= n;i ++) {
        scanf("%d",a+i);
    }
    scanf("%d",&m);
    for(int i = 1;i <= m;i ++) {
        scanf("%d%d",&qj[i].l,&qj[i].r);
        qj[i].id = i;
    }
    sort(qj + 1,qj + 1 + m,cmp);
    int nex = 1;
    for(int i = 1;i <= m;i ++) {
        for(;nex <= qj[i].r;nex ++) {
            if(las[a[nex]]) {
                add(las[a[nex]],-1);
            }
            add(nex,1);
            las[a[nex]] = nex;
        }
        ans[qj[i].id] = getsum(qj[i].r) - getsum(qj[i].l - 1);
    }
    for(int i = 1;i <= m;i ++) printf("%d\n",ans[i]);
    return 0;
}