BZOJ 1878 （主席树在线）

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

Day2

题目大意：询问区间不同数的个数。也可以用莫队来做主要就是练习一下主席树。

解题思路：对于一个区间里不同个数，如果查询的右端点固定我们就可以用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));
    }
}