[洛谷1972][SDOI2009]HH的项链

Posted 2020-10-14 lh

题目描述

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 <= 50000，M <= 200000。

莫队

用cnt[i]表示数i的出现次数，sum表示当前区间的数的种类，移动指针时，若cnt[i]==1，sum++，cnt[i]==0，sum--

cnt表示的是数的出现次数，不是数的下标的出现次数，容易误打

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath> 
using namespace std;
struct xxx{
    int l,r,block,id;
}q[200001];
int a[50001],cnt[1000100],ans[200001];
bool cmp(xxx a,xxx b){if(a.block!=b.block)return a.block<b.block;return a.r<b.r;}
int main()
{
    int n,m;scanf("%d",&n);int T=(int)sqrt((double)n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);q[i].block=(q[i].l+1)/T;q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    int l=1,r=0,sum=0;
    for(int i=1;i<=m;i++)
    {
        while(r<q[i].r){cnt[a[++r]]++;if(cnt[a[r]]==1)sum++;}
        while(r>q[i].r){cnt[a[r]]--;if(cnt[a[r--]]==0)sum--;}
        while(l<q[i].l){cnt[a[l]]--;if(cnt[a[l++]]==0)sum--;}
        while(l>q[i].l){cnt[a[--l]]++;if(cnt[a[l]]==1)sum++;}
        ans[q[i].id]=sum;
    }
    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
    return 0;
}