[Luogu P1972] [BZOJ 1878] [SDOI2009]HH的项链

洛谷传送门

BZOJ传送门

题目描述

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 <= 200000。

解题分析

原来的数据范围是$N\leq 50000, M\leq20000$， 还可以用莫队苟一苟， 但是$500000$嘛还是弃疗…

然而这道题有些很好的性质： 对于每个颜色我们实际上只需要考虑最靠近询问右边界的出现位置即可。例如252132， 如果查询区间是$[2,6]$的话， 我们只需要知道最后一个$2$的位置， 即$6$即可回答。

所以我们将询问按右端点排序， 树状数组维护每个数出现位置的贡献的前缀和即可。

代码如下：

#include <cstdio>
#include <cstdlib>
#include <cctype>
#include <cstring>
#include <cmath>
#include <algorithm>
#define R register
#define W while
#define gc getchar()
#define IN inline
#define MX 1000500
#define lbt(i) (i & -i)
template <class T>
IN void in(T &x)
{
    x = 0; R char c = gc;
    for (; !isdigit(c); c = gc);
    for (;  isdigit(c); c = gc)
    x = (x << 1) + (x << 3) + c - 48;
}
int t[MX], pre[MX], dat[MX], ans[MX];
struct Query {int lef, rig, id;} que[MX];
IN bool operator < (const Query &x, const Query &y)
{return x.rig < y.rig;}
int n, q;
IN void add(R int now, R int del)
{W (now <= n) t[now] += del, now += lbt(now);}
IN int query(R int now)
{
    int ret = 0;
    W (now) ret += t[now], now -= lbt(now);
    return ret;
}
int main(void)
{
    in(n);
    for (R int i = 1; i <= n; ++i) in(dat[i]);
    in(q);
    for (R int i = 1; i <= q; ++i) in(que[i].lef), in(que[i].rig), que[i].id = i;
    std::sort(que + 1, que + 1 + q);
    R int now = 0;
    for (R int i = 1; i <= q; ++i)
    {
        W (now < que[i].rig)
        {
            ++now;
            if(pre[dat[now]]) add(pre[dat[now]], -1);//之前未出现的数就不必消去了
            add(now, 1); pre[dat[now]] = now;
        }
        ans[que[i].id] = query(que[i].rig) - query(que[i].lef - 1);
    }
    for (R int i = 1; i <= q; ++i) printf("%d\n", ans[i]);
}