bzoj1878 [SDOI2009]HH的项链 主席树

Description

---

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

N ≤ 50000，M ≤ 200000。

Solution

---

如果m比较小其实可以裸上莫队，如果不强制在线可以离线然后树状数组
考虑强制在线怎么做。记last[i]为前一个与a[i]相等的位置，若不存在则为0。对于询问(L,R)我们统计的实际上是last[i]<L的数量，这个用主席树做就十分显然了

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

const int N=50005;
const int M=1000005;

struct treeNode {
	int l,r,sum;
} t[N*51];

int a[N],last[N],rec[M];
int root[N],tot;

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):v,ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void modify(int &now,int pre,int tl,int tr,int x) {
	t[now=++tot]=t[pre]; t[now].sum++;
	if (tl==tr) return ;
	int mid=(tl+tr)>>1;
	if (x<=mid) modify(t[now].l,t[pre].l,tl,mid,x);
	else modify(t[now].r,t[pre].r,mid+1,tr,x);
}

int query(int now,int pre,int tl,int tr,int l,int r) {
	if (tl==l&&tr==r) return t[now].sum-t[pre].sum;
	int mid=(tl+tr)>>1;
	if (r<=mid) return query(t[now].l,t[pre].l,tl,mid,l,r);
	if (l>mid) return query(t[now].r,t[pre].r,mid+1,tr,l,r);
	return query(t[now].l,t[pre].l,tl,mid,l,mid)+query(t[now].r,t[pre].r,mid+1,tr,mid+1,r);
}

int main(void) {
	int n=read();
	rep(i,1,n) {
		a[i]=read();
		last[i]=rec[a[i]];
		rec[a[i]]=i;
		modify(root[i],root[i-1],0,n,last[i]);
	}
	for (int T=read();T--;) {
		int l=read(),r=read();
		printf("%d\n", query(root[r],root[l-1],0,n,0,l-1));
	}
	return 0;
}