洛谷1972 [SDOI2009]HH的项链（链表+树状数组）

题目描述

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

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：

6
1 2 3 4 3 5
3
1 2
3 5
2 6

输出样例#1：

2
2
4

说明

数据范围：
对于100%的数据，N <= 50000，M <= 200000。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1000006;
struct data{
    int id,l,r;
    bool operator<(const data&b)const{
	    return (l<b.l ||(l==b.l && r<b.r));
	}
}q[maxn];
int n,m,a[maxn],fst[maxn],nxt[maxn],s[maxn],mx,ans[maxn];
inline int get(){
    char c;while(!isdigit(c=getchar()));
    int v=c-48;while(isdigit(c=getchar()))v=v*10+c-48;
    return v;
}
inline int lowbit(int x){return x&(-x);}
inline void updata(int x,int v){
    while(x<=n){
	    s[x]+=v;x+=lowbit(x);
	}
}
inline int getsum(int x){
	int res=0;
    while(x){
	    res+=s[x];x-=lowbit(x);
	}
	return res;
}
int main(){
    n=get();memset(s,0,sizeof(s));
    for(int i=1;i<=n;++i)a[i]=get(),mx=max(mx,a[i]);
    for(int i=n;i>=1;--i)nxt[i]=fst[a[i]],fst[a[i]]=i;
    for(int i=1;i<=mx;++i)if(fst[i])updata(fst[i],1);
    m=get();
    for(int i=1;i<=m;++i)q[i].l=get(),q[i].r=get(),q[i].id=i;
    sort(q+1,q+1+m);
    int lft=1;
    for(int i=1;i<=m;++i){
	    while(lft<q[i].l){
		    if(nxt[lft])updata(nxt[lft],1);
		    ++lft;
		}
		ans[q[i].id]=getsum(q[i].r)-getsum(q[i].l-1);
	}
	for(int i=1;i<=m;++i)printf("%d\n",ans[i]);
	return 0;
}

思路：树状数组离线。按左端点排序。初始将每个数第一个出现的都标记上1。以后的每个区间中的数只有两种情况，一种是第一次出现，一种是前面出现过的。对于前者已经处理好了，后者用链表维护，即在当前查询区间左端点

之前出现的x都s[ nxt[ x ] ]++。最后输出sum[ r ] - sum[ l-1 ]即可。

        因为之前的x若在当前区间出现，s[ nxt[ x ] ]++，即求和之后相减能够保留且只保留一次x。