[BZOJ 2743] [HEOI 2012] 采花

Description

萧芸斓是Z国的公主，平时的一大爱好是采花。今天天气晴朗，阳光明媚，公主清晨便去了皇宫中新建的花园采花。花园足够大，容纳了 \(n\) 朵花，花有 \(c\) 种颜色（用整数 \(1-c\) 表示），且花是排成一排的，以便于公主采花。公主每次采花后会统计采到的花的颜色数，颜色数越多她会越高兴！同时，她有一癖好，她不允许最后自己采到的花中，某一颜色的花只有一朵。为此，公主每采一朵花，要么此前已采到此颜色的花，要么有相当正确的直觉告诉她，她必能再次采到此颜色的花。由于时间关系，公主只能走过花园连续的一段进行采花，便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了 \(m\) 个行程，然后一一向你询问公主能采到多少朵花（她知道你是编程高手，定能快速给出答案！），最后会选择令公主最高兴的行程（为了拿到更多奖金！）。

Input

第一行四个空格隔开的整数 \(n,c\) 以及 \(m\)。

接下来一行 \(n\) 个空格隔开的整数，每个数在 \([1, c]\) 间，第 \(i\) 个数表示第 \(i\) 朵花的颜色。

接下来 \(m\) 行每行两个空格隔开的整数 \(l\) 和 \(r\) \((l ≤ r)\)，表示女仆安排的行程为公主经过第 \(l\) 到第 \(r\) 朵花进行采花。

Output

共 \(m\) 行，每行一个整数，第 \(i\) 个数表示公主在女仆的第 \(i\) 个行程中能采到的花的颜色数。

Sample Input

5 3 5
1 2 2 3 1
1 5
1 2
2 2
2 3
3 5

Sample Output

2 
0 
0 
1 
0 

HINT

\(1 ≤ n ≤ 10^6,c ≤ n,m ≤10^6\)

Solution

HH的项链加强版。将询问按照左端点排序，设 \(nxt[i]\) 表示下一个与 \(i\) 颜色相同的点的位置。

当扫过点 \(i\) 后，HH的项链是 update(i, -1), update(nxt[i], 1)，这里改成 update(nxt[i], -1), update(nxt[nxt[i]], 1) 就好。

Code

#include <cstdio>
#include <algorithm>

const int N = 1000005;
struct Node {
    int l, r, id;
    bool operator < (const Node &rhs) const {
        return l < rhs.l;
    }
} b[N];
int nxt[N], pos[N], sum[N], ans[N], a[N], n, c, m;

int read() {
    int x = 0; char c = getchar();
    while (c < '0' || c > '9') c = getchar();
    while (c >= '0' && c <= '9') x = (x << 3) + (x << 1) + (c ^ 48), c = getchar();
    return x;
}
void update(int x, int y) {
    while (x <= n) sum[x] += y, x += x & (-x);
}
int query(int x) {
    int res = 0;
    while (x) res += sum[x], x -= x & (-x);
    return res;
}
int main() {
    n = read(), c = read(), m = read();
    for (int i = 1; i <= n; ++i) a[i] = read();
    for (int i = n; i; --i) {
        if (pos[a[i]]) nxt[i] = pos[a[i]];
        pos[a[i]] = i;
    }
    for (int i = 1; i <= m; ++i) b[i].l = read(), b[i].r = read(), b[i].id = i;
    std::sort(b + 1, b + m + 1);
    for (int i = 1; i <= c; ++i) if (nxt[pos[i]]) update(nxt[pos[i]], 1);
    for (int i = 1, j = 1; i <= n; ++i) {
        while (b[j].l == i) ans[b[j].id] = query(b[j].r), ++j;
        if (nxt[i]) {
            update(nxt[i], -1);
            if (nxt[nxt[i]]) update(nxt[nxt[i]], 1);
        }
    }
    for (int i = 1; i <= m; ++i) printf("%d\n", ans[i]);
    return 0;
}

标签：nxt,颜色,HEOI,采花,采到,int,2743,公主,2012
来源： https://www.cnblogs.com/fly-in-milkyway/p/10366827.html