洛谷 P1972 [SDOI2009]HH的项链

P1972 [SDOI2009]HH的项链

题目背景

无

题目描述

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。

思路：莫队。

洛谷：AC。cogs：WA+TLE，数组开小了。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 50001
#define M 200001
#define MAXN 1000001
using namespace std;
int n,m,S,ans;
int num[N],sum[MAXN];
struct nond{
    int l,r,id,pos,ans;
}edge[M];
int cmp1(nond a,nond b){
    return a.id<b.id;
}
int cmp(nond a,nond b){
    if(a.pos==b.pos)    return a.r<b.r;
    else return a.pos<b.pos;
}
void up(int x,int k){
    if(k==1&&!sum[num[x]])    ans++;
    if(k==-1&&sum[num[x]]==1)    ans--;
    sum[num[x]]+=k;
}
void mode(){
    int l=1,r=0;
    for(int i=1;i<=m;i++){
        while(l<edge[i].l)    up(l++,-1);
        while(l>edge[i].l)    up(--l,1);
        while(r<edge[i].r)    up(++r,1);
        while(r>edge[i].r)    up(r--,-1);
        edge[i].ans=ans;
    }
}
int main(){
    freopen("diff.in","r",stdin);
    freopen("diff.out","w",stdout);
    scanf("%d",&n);
    S=sqrt(n);
    for(int i=1;i<=n;i++)    scanf("%d",&num[i]);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&edge[i].l,&edge[i].r);
        edge[i].id=i;
        edge[i].pos=(edge[i].l-1)/S+1;
    }
    sort(edge+1,edge+1+m,cmp);
    mode();
    sort(edge+1,edge+1+m,cmp1);
    for(int i=1;i<=m;i++)
        cout<<edge[i].ans<<endl;
}