BZOJ1878: [SDOI2009]HH的项链

BZOJ1878: [SDOI2009]HH的项链

题目描述

HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运，所以每次散步 完后，他都会随意取出一
段贝壳，思考它们所表达的含义。HH不断地收集新的贝壳，因此他的项链变得越来越长。有一天，他突然提出了一
个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答。。。因为项链实在是太长了。于是，他只
好求助睿智的你，来解决这个问题。

输入

第一行：一个整数N，表示项链的长度。
第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。
第三行：一个整数M，表示HH询问的个数。
接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。
N ≤ 50000，M ≤ 200000。

输出

M行，每行一个整数，依次表示询问对应的答案。

---

Solution

离线+树状数组
首先将所有询问离线,按照询问的左区间排序
然后考虑每一个数,能够对答案产生贡献的区间是在这个数的位置到下一个和这个一样的数的位置,因为如果两个数在同一个区间里,他会被计算两次
记录一下每个数下一次出现的位置
接下来既然考虑了离线,就要想到从左往右扫描,然后求前缀和
对于每一个扫描到的点,它右边的每个数最多被记录了1次,而左边的每个数被记录了若干次
所以前缀和一减就是答案
可能比较玄学,不过代码还是比较好理解的

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 50010
#define maxm 200010
#define maxr 1000010
inline int read(){
    int ret=0,ff=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') ff=-ff;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        ret=ret*10+ch-'0';
        ch=getchar();
    }
    return ret*ff;
}
struct X{
    int l,r,ans,id;
}q[maxm];
struct cmp1{
    bool operator()(const X& t1,const X& t2){
        return t1.l<t2.l;
    }
};
struct cmp2{
    bool operator()(const X& t1,const X&t2){
        return t1.id<t2.id;
    }
};
int a[maxn],nxt[maxn],p[maxr],tr[maxn];
void add(int x,int val,int n){
    while(x<=n){
        tr[x]+=val;
        x+=x&-x;
    }
}
int query(int x){
    int ret=0;
    while(x){
        ret+=tr[x];
        x-=x&-x;
    }
    return ret;
}
int main(){
    freopen("bzoj1878.in","r",stdin);
    int n=read(),maxx=0;
    for(int i=1;i<=n;++i) a[i]=read(),maxx=max(maxx,a[i]);
    for(int i=n;i;--i) nxt[i]=p[a[i]],p[a[i]]=i;
    for(int i=0;i<=maxx;++i)if(p[i]) add(p[i],1,n);//先对每一个数第一次出现的位置做前缀和
    int m=read();
    for(int i=1;i<=m;++i) q[i].l=read(),q[i].r=read(),q[i].id=i;
    sort(q+1,q+1+m,cmp1());//离线按左端点排序
    int L=1;
    for(int i=1;i<=m;++i){
        while(L<q[i].l){
            if(nxt[L])add(nxt[L],1,n);
            //如果询问的左端点已经超过了当前扫描到的点,之前做的前缀和已经失去意义
            //有意义的点应该在这个数下一次出现的位置上
            ++L;
        }
        q[i].ans=query(q[i].r)-query(L-1);
    }
    sort(q+1,q+1+m,cmp2());
    for(int i=1;i<=m;++i) printf("%d\n",q[i].ans);
    return 0;
}