落谷p4113采花

落谷题目链接

题目描述

萧薰儿是古国的公主，平时的一大爱好是采花。

今天天气晴朗，阳光明媚，公主清晨便去了皇宫中新建的花园采花。

花园足够大，容纳了n朵花，花有c种颜色（用整数1-c表示），且花是排成一排的，以便于公主采花。公主每次采花后会统计采到的花的颜色数，颜色数越多她会越高兴！同时，她有一癖好，她不允许最后自己采到的花中，某一颜色的花只有一朵。为此，公主每采一朵花，要么此前已采到此颜色的花，要么有相当正确的直觉告诉她，她必能再次采到此颜色的花。

由于时间关系，公主只能走过花园连续的一段进行采花，便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了m个行程，然后一一向你询问公主能采到多少朵花（她知道你是编程高手，定能快速给出答案！），最后会选择令公主最高兴的行程（为了拿到更多奖金！）。

输入格式

第一行四个空格隔开的整数n、c以及m。接下来一行n个空格隔开的整数，每个数在[1, c]间，第i个数表示第i朵花的颜色。接下来m行每行两个空格隔开的整数l和r（l ≤ r），表示女仆安排的行程为公主经过第l到第r朵花进行采花。

输出格式

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

输入

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

输出

2
0
0
1
0

说明 提示

对于100%的数据，$1\le n\le 2\ast 10^6$ , $c\le n，m\le 2\ast 10^6.$
本题有两个subtask
subtask1保证$n,m,c\le 3\ast 10^5$，占100分
subtask2保证$n,m,c\le 2\ast 10^6$ ，占100分

思路

这道题用离线+树状数组可以过，我们可以看出来只有在一种颜色在第二次出现的时候，答案才会改变。用next数组记录第i个位置的颜色下一次出现的位置，nnext数组记录第i个位置下下次出现的位置，把每一种颜色第二次出现的时候将那个点赋值为一，把查询的左端点从小到大排序，然后维护左端点，当i位置不在查询区间内的时候，那么我们之前一定将他的下一个出现位置的值赋值了一，所以要减一，那么下下个出现的位置一定要+1，只用查询右端点的到值就行了。（用树状数组维护，应该可以用线段树，没试过用线段树）
平常的莫队可以过100的数据，优化的应该可以过133；
下面放代码（代码丑，如果理解了就不用看了）

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e6 + 10;
int c[maxn], color[maxn], next[maxn], nnext[maxn];
int n, m, C;
int lowbit(int x){
    return x & (-x);
}
int query(int x){
    int ans = 0;
    while(x){
        ans += c[x];
        x -= lowbit(x);
    }
    return ans;
}

void update(int x, int y){
    for(int i = x; i <= n; i += lowbit(i)){
        c[i] += y;
    }
}
struct Node
{
    int l, r, id;
}aa[maxn];
bool cmp(Node a, Node b)
{
    return a.l < b.l;
}
int frist[maxn], bb[maxn], ans[maxn];
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) {w|=ch=='-';ch=getchar();}
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
int main()
{
    n = read();
    C = read();
    m = read();
    for(int i = 1; i <= n; i++){
        color[i] = read();
    }
    for(int i = n; i >= 1; i--){
        next[i] = frist[color[i]];
        frist[color[i]] = i;
    }
    for(int i = 1; i <= n; i++){
        nnext[i] = next[next[i]];
    }
    for(int i = 1; i <= m; i++){
        aa[i].l = read();
        aa[i].r = read();
        aa[i].id = i;
    }
    for(int i = 1; i <= n; i++){
        if(++bb[color[i]] == 2) update(i, 1);
    }
    sort(aa + 1, aa + 1 + m, cmp);
    int now = 1;
    for(int i = 1; i <= m; i++){
        for(now; now <= aa[i].l - 1; now++){
            if(next[now]) update(next[now], -1);
            if(nnext[now]) update(nnext[now], 1);
        }
        ans[aa[i].id] = query(aa[i].r);
    }
    for(int i = 1; i <= m; i++){
        printf("%d\n", ans[i]);
    }
    return 0;
}