洛谷P1972 [SDOI2009]HH的项链 #树状数组 离线操作#

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。

有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答…… 因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入格式

一行一个正整数 nn，表示项链长度。
第二行 nn 个正整数 a_iai​，表示项链中第 ii 个贝壳的种类。

第三行一个整数 mm，表示 HH 询问的个数。
接下来 mm 行，每行两个整数 l,rl,r，表示询问的区间。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

2
2
4

说明/提示

【数据范围】

对于 20\%20% 的数据，1\le n,m\leq 50001≤n,m≤5000；
对于 40\%40% 的数据，1\le n,m\leq 10^51≤n,m≤105；
对于 60\%60% 的数据，1\le n,m\leq 5\times 10^51≤n,m≤5×105；
对于 100\%100% 的数据，1\le n,m,a_i \leq 10^61≤n,m,ai​≤106，1\le l \le r \le n1≤l≤r≤n。

本题可能需要较快的读入方式，最大数据点读入数据约 20MB

题解

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

typedef pair<int, int> p;
const int maxn = 1e6 + 10;
int n, m, a[maxn], c[maxn], ans[maxn], last[maxn];
struct que { int l, r, id; };
vector<que> vec;

inline const int read()
{
    int x = 0, f = 1; char ch = getchar();
    while (ch < '0' || ch > '9') { if (ch == '-') f = -1; ch = getchar(); }
    while (ch >= '0' && ch <= '9') { x = (x << 3) + (x << 1) + ch - '0'; ch = getchar(); }
    return x * f;
}

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

void update(int p, int v) { for (int i = p; i <= n; i += lowbit(i)) c[i] += v; }

int getSum(int p)
{
    int res = 0;
    for (int i = p; i > 0; i -= lowbit(i)) res += c[i];
    return res;
}

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

int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("input.txt", "r", stdin);
#endif
    n = read();
    for (int i = 1; i <= n; i++) a[i] = read();
    m = read();
    for (int i = 1; i <= m; i++)
    {
        int l = read(), r = read();
        vec.push_back(que{ l, r, i });
    }
    sort(vec.begin(), vec.end(), cmp);
    int now = 1;
    for (auto q : vec)
    {
        for (int i = now; i <= q.r; i++)
        {
            if (last[a[i]]) update(last[a[i]], -1);
            last[a[i]] = i;
            update(i, 1);
        }
        now = q.r;
        ans[q.id] = getSum(q.r) - getSum(q.l - 1);
    }
    for (int i = 1; i <= m; i++) printf("%d\n", ans[i]);
    return 0;
}