bzoj-1878 HH的项链

标签：bzoj   树状数组   

题意：

给定长为n的序列与m个询问；

求询问的区间里不相同数字的个数；

n<=50000,m<=200000,数字<=1000000

题解：

在线算法太过高深并不会，所以这道题就要离线做；

分析数据范围，m个询问完全可以存下，1000000的数字也可以hash勉强不用离散化；

那么考虑一段区间的数字种类数，大概就是 1 - R 减去 1 - ( L - 1 ) ；

但是因为有重复的数字，对于重复了的数，我们只应该在L-R间计算；

那就在遍历询问时顺便记录数组pre[ x ]，表示x这个数字的前一个在什么位置(初值为0即可)；

用树状数组维护数字种类数的前缀和，遇到记录过的数字就把前一个删掉，加入新的；

 然后以右端点排序询问，每遇到右端点就可以解决询问；

没什么坑，想明白了就1A；

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 50005
#define M 200002
#define V 1000001
using namespace std;
int a[N],pre[V],tree[N];
struct node
{
	int l,r,ans,num;
}q[M];
int cmp(node a,node b)
{
	return a.r<b.r;
}
int cmp2(node a,node b)
{
	return a.num<b.num;
}
int lowbit(int k)
{
	return k&(-k);
}
void add(int k,int val)
{
	while(k<=N)
	{
		tree[k]+=val;
		k+=lowbit(k);
	}
}
int query(int k)
{
	int ret=0;
	while(k)
	{
		ret+=tree[k];
		k-=lowbit(k);
	}
	return ret;
}
int main()
{
	int n,m,i,j,k,l,r;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%d",a+i);
	scanf("%d",&m);
	for(i=1;i<=m;i++)
		scanf("%d%d",&q[i].l,&q[i].r),q[i].num=i;
	sort(q+1,q+1+m,cmp);
	for(i=1,j=1;i<=n;i++)
	{
		if(pre[a[i]])
			add(pre[a[i]],-1);
		add(i,1);
		pre[a[i]]=i;
		if(i==q[j].r)	k=query(i);
		while(i==q[j].r)
		{
			q[j++].ans=k-query(q[j].l-1);
		}
	}
	sort(q+1,q+1+m,cmp2);
	for(i=1;i<=m;i++)
	{
		printf("%d\n",q[i].ans);
	}
	return 0;
}

bzoj-1878 HH的项链

标签：bzoj   树状数组   

原文地址：http://blog.csdn.net/ww140142/article/details/45461225