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

AC代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;

const int N = 5e5 + 5;
struct Node {
    int l, r, sww, num;
    Node () {}
    Node (int l, int r, int sww, int num) : l (l), r (r), sww (sww), num (num) {}
}q[N];
int n, m, pos = 1;
int a[N], vis[N << 1], sum[N];

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

void modify (int o, int val) {
    while (o <= n) {
        sum[o] += val;
        o += lowbit (o);
    }
    return ;
}

int query (int o) {
    int tot = 0;
    while (o > 0) {
        tot += sum[o];
        o -= lowbit (o);
    }
    return tot;
}

bool cmp_1 (Node a, Node b) {
    return a.r < b.r;
}

bool cmp_2 (Node a, Node b) {
    return a.num < b.num;
}

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++) {
        int l, r;
        scanf ("%d%d", &l, &r);
        q[i] = Node (l, r, 0, i);
    }
    sort (q + 1, q + m + 1, cmp_1);
    for (int i = 1; i <= n; i++) {
        if (vis[a[i]]) modify (vis[a[i]], -1);
        vis[a[i]] = i, modify (i, 1);
        while (q[pos].r == i) {
            q[pos].sww = query (q[pos].r) - query (q[pos].l - 1);
            pos++;
        }
        if (pos > m) break;
    }
    sort (q + 1, q + m + 1, cmp_2);
    for (int i = 1; i <= m; i++) printf ("%d\n", q[i].sww);
    return 0;
}