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