1878: [SDOI2009]HH的项链
Time Limit: 4 Sec Memory Limit: 64 MB
Submit: 7107 Solved: 3504
[Submit][Status][Discuss]
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
HINT
Source
题目大意:询问区间不同数的个数。也可以用莫队来做 主要就是练习一下主席树。
解题思路:对于一个区间里不同个数,如果查询的右端点固定我们就可以用s[r]-s[l-1]来得到答案(同一数字在最右端的是有效的)。
例如: 1 2 3 3 4 5
数组s: 1 2 2 3 4 5
这样对于固定右端点的查询就可做了,恰好主席树就是一个记录数组前缀的树。对于要查询的区间l到r我们在第r棵树上查询l到r即为答案
#include <iostream>
#include <string>
#include<cstdio>
#include<cstring>
#include <algorithm>
using namespace std;
#define LL long long
#define N 50005
#define rep(i,j,k) for(int i=j;i<=k;i++)
int sum[N*30],L[N*30],R[N*30],T[N*30];
int tot=0;
int a[N],b[N];
void build(int &rt,int l,int r)
{
rt=++tot;
sum[rt]=0;
if(l==r)return ;
int m=(l+r)>>1;
build(L[rt],l,m);
build(R[rt],m+1,r);
}
void update(int &now,int pre,int l,int r,int p,int val)
{
now=++tot;
L[now]=L[pre];
R[now]=R[pre];
sum[now]=sum[pre]+val;
if(l==r)return ;
int m=(l+r)>>1;
if(p<=m)update(L[now],L[pre],l,m,p,val);
else update(R[now],R[pre],m+1,r,p,val);
}
int query(int rt,int ql,int qr,int l,int r)
{
if(l>=ql && r<=qr)return sum[rt];
int m=(l+r)>>1;
int ans=0;
if(ql<=m) ans+=query(L[rt],ql,qr,l,m);
if(qr>m) ans+=query(R[rt],ql,qr,m+1,r);
return ans;
}
int vis[1000005];
int main()
{
int n;
scanf("%d",&n);
rep(i,1,n)scanf("%d",&a[i]),b[i]=a[i];
memset(vis,0,sizeof(vis));
build(T[0],1,n);
for(int i=1;i<=n;i++)
{
if(vis[a[i]]==0) update(T[i],T[i-1],1,n,i,1);
else {
int tmp;
update(T[i],T[i-1],1,n,vis[a[i]],-1);
update(T[i],T[i],1,n,i,1);
}
vis[a[i]]=i;
}
int m;
scanf("%d",&m);
rep(i,1,m)
{
int l,r;
scanf("%d%d",&l,&r);
printf("%d\n",query(T[r],l,r,1,n));
}
}