P1972 [SDOI2009]HH的项链

题目背景

无

题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH 不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答……因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

输入格式：

第一行：一个整数N，表示项链的长度。

第二行：N 个整数，表示依次表示项链中贝壳的编号（编号为0 到1000000 之间的整数）。

第三行：一个整数M，表示HH 询问的个数。

接下来M 行：每行两个整数，L 和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

输出格式：

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

输入样例：

6

1 2 3 4 3 5

3

1 2

3 5

2 6

输出样例：

2

2

4

数据范围：

对于100%的数据，N <= 500000，M <= 500000。

思路：

莫队一定要加优化！！！

代码：

// luogu-judger-enable-o2
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int N=1010000;

int aa[N],cnt[N],belong[N];
int n,m,size,bnum,now,ans[N];

struct node {
    int l, r, id;
} q[N];

bool cmp(node a,node b) {
    if(belong[a.l]==belong[b.l]) {
        if(belong[a.l]%2)
            return a.r<b.r;
        else
            return a.r>b.r;
    }
    return belong[a.l]<belong[b.l];
}

inline int read() {
    int s=0,w=1;
    char ch=getchar();
    while(ch<'0'||ch>'9') {
        if(ch=='-')
            w=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9') {
        s=s*10+ch-'0';
        ch=getchar();
    }
    return s*w;
}

void print(int x) {
    if(x/10)
        print(x/10);
    putchar(x%10+'0');
}

int main() {
    int i,j;
    scanf("%d",&n);
    size=sqrt(n);
    bnum=ceil(n*1.0/size);
    for(i=1; i<=bnum; i++)
        for(j=(i-1)*size+1; j<=i*size; j++)
            belong[j]=i;
    for(i = 1; i <= n; ++i)
        aa[i]=read();
    m=read();
    for(i=1; i<=m; i++) {
        q[i].l=read();
        q[i].r=read();
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    int l=1,r=0;
    for(i=1; i<=m; i++) {
        int ql=q[i].l,qr=q[i].r;
        while(l>ql)
            now+=!cnt[aa[--l]]++;
        while(r<qr)
            now+=!cnt[aa[++r]]++;
        while(l<ql)
            now-=!--cnt[aa[l++]];
        while(r>qr)
            now-=!--cnt[aa[r--]];
        ans[q[i].id] = now;
    }
    for(i=1; i<=m; i++) {
        print(ans[i]);
        putchar('\n');
    }
    return 0;
}