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;
}

 

上一篇:LGOJ P4137 Rmq Problem / mex


下一篇:牛客提高D4t3 清新题