[BZOJ1878]SDOI2009HH的项链|树状数组

这道题纠结了好久。。只想得到线段树套平衡树的方法。。莫队会T掉。。后来看代码终于想通了，还是要离线搞。。读入之后预处理出每个位置下一个同颜色的位置next[]，再将询问按左端点升序排序，用sum[i]表示1-i的颜色数，初始时只记录每种颜色第一个出现的位置。。用一个l指针，表示所有l之前的所有位置的颜色都已用next[]转移到l后了，这样保证颜色不重不漏，依次处理询问，将l不断向右移动，移动时把l位置的颜色在sum中去掉，再将next[l]在sum中加上，用树状数组维护前缀和，每次询问的答案就是sum(q.r)-sum(q.l-1)。（真心不会讲，要意会，还是看代码好。。）

#include<cstdio>
#include<iostream>
#include<memory.h>
#include<cmath>
#include<algorithm>
using namespace std;
struct query{
	int l,r,y;
} q[200005];
int n,m,i,l,maxcol=0,head[1000005],col[50005],c[50005],next[50005],ans[200005];
bool cmp(query a,query b)
{
	return a.l<b.l;
}
int lowbit(int x) {return x&-x;}
void add(int x,int k)
{
	for (;x<=n;x+=lowbit(x)) c[x]+=k;
}
int getsum(int x)
{
	int ans=0;
	for (;x;x-=lowbit(x)) ans+=c[x];
	return ans;
}
int main()
{
	freopen("1878.in","r",stdin);
	scanf("%d",&n);
	for (i=1;i<=n;i++) scanf("%d",&col[i]);
	scanf("%d",&m);
	for (i=1;i<=m;i++)
	{
		scanf("%d%d",&q[i].l,&q[i].r);
		q[i].y=i;
	}
	sort(q+1,q+1+m,cmp);
	memset(head,0,sizeof(head));
	for (i=n;i;i--)
	{
		next[i]=head[col[i]];
		head[col[i]]=i;
		maxcol=max(maxcol,col[i]);
	}
	memset(c,0,sizeof(c));
	for (i=1;i<=maxcol;i++) if (head[i]) add(head[i],1);
	l=1;
	for (i=1;i<=m;i++)
	{
		while (l<q[i].l) if (next[l++]) add(next[l-1],1);
		ans[q[i].y]=getsum(q[i].r)-getsum(q[i].l-1);
	}
	for (i=1;i<=m;i++) printf("%d\n",ans[i]);
}