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