树桩数组维护区间不同值的个数

题目

原本想用莫队水一水

结果只有80分,不够学了一招。

将问题转化为离线问题。

我们先将询问按右节点升序排序。

然后对于每一个值,我们可以肯定的是,只有最靠近询问右端的值才可能起作用。

case 1 :如果询问不包括某个值。那么这个值,对答案没有影响

case 2 :包括的话,那么至少包括1个这样的值,而我们询问的是不同的个数,所以对答案也没有影响

这样的话,我们只需要维护最靠右的位置在哪就行了。

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
struct BIT
{
	int num;
	int data[1000100];
	void insert(int value,int pos)
	{
		while(pos<=num)
		{
			data[pos]+=value;
			pos+=(pos&(-pos));
		}
	}
	int sum(int pos)
	{
		int ans=0;
		while(pos)
		{
			ans+=data[pos];
			pos-=(pos&(-pos));
		}
		return ans;
	}
	int check(int l,int r)
	{
		return sum(r)-sum(l-1);
	}
};
BIT bit;
int last[1000100];
int base[500100];
int ans[200100];
struct node
{
	int left;
	int right;
	int num;
};
bool compare(const node &a,const node &b)
{
	if(a.right==b.right)
		return a.left<b.left;
	return a.right<b.right;
}
int read()
{
	int s=0,f=1;
	char in=getchar();
	while(in<'0'||in>'9')
	{
		if(in=='-')
			f=-1;
		in=getchar();
	}
	while(in>='0'&&in<='9')
	{
		s=(s<<1)+(s<<3)+in-'0';
		in=getchar();
	}
	return s*f;
}
node query[200100];
int main()
{
	int n=read();
	for(int i=1;i<=n;i++)
		base[i]=read();
	bit.num=n;
	int m=read();
	for(int i=1;i<=m;i++)
	{
		query[i].left=read();
		query[i].right=read();
		query[i].num=i;//由于是离线操作吗——
	}
	sort(query+1,query+1+m,compare);//排序,左端点其实可以不拍
	int i=0;//i是当前遍历到的右端点
	for(int j=1;j<=m;j++)
	{
		while(i<query[j].right)
		{
			i+=1;
			if(last[base[i]])
			{
				bit.insert(-1,last[base[i]]);
				bit.insert(1,i);
				last[base[i]]=i;
			}
			else
			{
				bit.insert(1,i);
				last[base[i]]=i;
			}
		}
		//printf("%d\n",bit.check(query[i].left,query[i].right));
		ans[query[j].num]=bit.check(query[j].left,query[j].right);
	}
	for(i=1;i<=m;i++)
		printf("%d\n",ans[i]);
}
posted @ 2018-04-23 21:09  Lance1ot  阅读(189)  评论(0编辑  收藏  举报