1878: [SDOI2009]HH的项链
Time Limit: 4 Sec Memory Limit: 64 MB
Submit: 7291 Solved: 3600
Description
HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运,所以每次散步 完后,他都会随意取出一
段贝壳,思考它们所表达的含义。HH不断地收集新的贝壳,因此他的项链变得越来越长。有一天,他突然提出了一
个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答。。。因为项链实在是太长了。于是,他只
好求助睿智的你,来解决这个问题。
Input
第一行:一个整数N,表示项链的长度。
第二行:N个整数,表示依次表示项链中贝壳的编号(编号为0到1000000之间的整数)。
第三行:一个整数M,表示HH询问的个数。
接下来M行:每行两个整数,L和R(1 ≤ L ≤ R ≤ N),表示询问的区间。
N ≤ 50000,M ≤ 200000。
Output
M行,每行一个整数,依次表示询问对应的答案。
Sample Input
6
1 2 3 4 3 5
3
1 2
3 5
2 6
Sample Output
2
2
4
解析:
主席树/莫队。
主席树的思路有点妙啊,对于一个询问,我们只需要知道有多少个数前一个出现的位置小于L,预处理出 last[ i ],然后主席树解决。
代码:
#include <bits/stdc++.h>
using namespace std;
const int Max=50005;
int n,m,tot,top;
int last[Max],pos[Max],a[Max],num[Max];
int tree[Max*20],lc[Max*20],rc[Max*20],rt[Max*20];
inline int get_int()
{
int x=0,f=1;char c;
for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
if(c=='-') f=-1,c=getchar();
for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
return x*f;
}
inline void print(int x)
{
if(x>9) print(x/10);
putchar('0'+x%10);
}
inline void add(int fa,int &now,int l,int r,int pos)
{
now=++tot,tree[now]=tree[fa]+1,lc[now]=lc[fa],rc[now]=rc[fa];
if(l==r) return;
int mid=(l+r)>>1;
if(pos<=mid) add(lc[fa],lc[now],l,mid,pos);
else add(rc[fa],rc[now],mid+1,r,pos);
}
inline int Q(int rt1,int rt2,int l,int r,int L,int R)
{
if(L<=l&&R>=r) return tree[rt1]-tree[rt2];
int mid=(l+r)>>1,ans=0;
if(L<=mid) ans+=Q(lc[rt1],lc[rt2],l,mid,L,R);
if(R>mid) ans+=Q(rc[rt1],rc[rt2],mid+1,r,L,R);
return ans;
}
int main()
{
n=get_int();
for(int i=1;i<=n;i++) num[i]=a[i]=get_int();
sort(a+1,a+n+1);
top=unique(a+1,a+n+1)-a-1;
for(int i=1;i<=n;i++) num[i]=lower_bound(a+1,a+top+1,num[i])-a,last[i]=pos[num[i]],pos[num[i]]=i;
for(int i=1;i<=n;i++) add(rt[i-1],rt[i],0,n,last[i]);
m=get_int();
while(m--)
{
int l=get_int(),r=get_int();
print(Q(rt[r],rt[l-1],0,n,0,l-1));
if(m) putchar('\n');
}
return 0;
}