[树状数组(离线)]BZOJ 1878: [SDOI2009]HH的项链 题解

（传送门）

题目大意

给出一个序列，求对于给定询问的区间中有多少个不同的数。

解题分析

离线求法，假设目前枚举的答案左端点为L(L从左到右枚举)，那么

对于一个数，在序列中会出现多次，但是我们只需要考虑最左边的一个

定义数组nxt，nxt[i]表示在i之后第一个与a[i]相同的数的位置。
当L往后推时，我们要始终维护：从L开始，各种数字第一个出现的位置记为+1，其他的位置都为0。
当L这个位置即将退出时，就把L位置-1，nxt[L]位置+1。
这样利用树状数组，求一趟前缀和sum(R)即可。

#include<cstdio>
#include<cstring>
#define maxn 50005
#define maxm 200005
#define maxv 1000005
using namespace std;
int n,q,tot,nxt[maxm],id[maxm],lnk[maxn],a[maxn],lst[maxv],ner[maxn],tee[maxn];
struct data{
    int L,R,ans;
    data (int L=0,int R=0):L(L),R(R){ }
}b[maxm];
inline void readi(int &x){
    x=0; char ch=getchar();
    while ('0'>ch||ch>'9') ch=getchar();
    while ('0'<=ch&&ch<='9') {x=x*10+ch-'0'; ch=getchar();}
}
void _change(int i,int val){for (;i<=n;i+=(i&(-i))) tee[i]+=val;}
int _get(int x){
    int tem=0; for (;x;x-=(x&(-x))) tem+=tee[x]; return tem;
}
void _add(int x,int y){
    id[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;
}
void _init(){
    freopen("necklace.in","r",stdin);
    freopen("necklace.out","w",stdout);
    memset(lnk,0,sizeof(lnk));
    memset(nxt,0,sizeof(nxt));
    readi(n); for (int i=1;i<=n;i++) readi(a[i]);
    readi(q); tot=0; for (int i=1,L,R;i<=q;i++) {readi(L); readi(R); b[i]=data(L,R); _add(L,i);};
}
void _solve()
{
    memset(lst,0,sizeof(lst));
    memset(ner,255,sizeof(ner));
    for (int i=1;i<=n;i++){
        if(lst[a[i]]) ner[lst[a[i]]]=i; else _change(i,1);
        lst[a[i]]=i;
    }
    for (int i=1;i<=n;i++){
        for (int j=lnk[i];j;j=nxt[j]) b[id[j]].ans=_get(b[id[j]].R);
        _change(i,-1); if (ner[i]!=-1) _change(ner[i],1);
    }
    for (int i=1;i<=q;i++) printf("%d\n",b[i].ans);
}
int main()
{
    _init();
    _solve();
    return 0;
}