【bzoj 1878】[SDOI2009]HH的项链

1878: [SDOI2009]HH的项链

Time Limit: 4 Sec   Memory Limit: 64 MB
Submit: 3905   Solved: 1953
[ Submit][ Status][ Discuss]

Description

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

Input

第一行：一个整数N，表示项链的长度。 第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。 第三行：一个整数M，表示HH询问的个数。 接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

Output

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

Sample Input

6
1 2 3 4 3 5
3
1 2
3 5
2 6

Sample Output

2
2
4

感觉到莫队算法的恐怖了

#include<bits/stdc++.h>
using namespace std;
const int maxn = 50050;
const int maxm = 200005;
struct data{int l,r,id,ans;}a[maxm];
int n,m,block,c[1000005],pos[maxn],in[maxn],ans,ret;
bool cmp(data _a,data _b)
{
    if(pos[_a.l] == pos[_b.l])return _a.r<_b.r;
    return _a.l<_b.l;
}
bool cmp_id(data _a,data _b)
{
    return _a.id<_b.id;
}
void init()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",in+i);
    block=(int)sqrt(n);
    for(int i=1;i<=n;i++)
        pos[i]=(i-1)/block+1;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a[i].l,&a[i].r);
        a[i].id=i;
    }
}
void update(int loc,int flag)
{
    //cout<<loc<<" "<<flag<<endl;
    if(flag==1){
        if(c[in[loc]]==0)ret++;
        c[in[loc]]++;
    }else{
        c[in[loc]]--;
        if(c[in[loc]]==0)ret--;
    }
}
void solve()
{
    for(int i=1,l=0,r=0;i<=m;i++){
        for(;r<a[i].r;r++)update(r+1,1);
        for(;r>a[i].r;r--)update(r,-1);
        for(;l<a[i].l;l++)update(l,-1);
        for(;l>a[i].l;l--)update(l-1,1); 
        a[i].ans=ret;   
    }
}
int main()
{
    init();
    sort(a+1,a+1+m,cmp);
    solve();
    sort(a+1,a+1+m,cmp_id);
    for(int i=1;i<=m;i++)
        printf("%d\n",a[i].ans);
    return 0;
}