求区间不同数的个数【树状数组求解】

最新推荐文章于 2025-04-26 21:39:52 发布

原创最新推荐文章于 2025-04-26 21:39:52 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

118 篇文章

订阅专栏

本文介绍了一种使用树状数组解决区间查询问题的方法。通过离线处理多个查询请求，该算法可以有效地更新和查询区间内元素的累计和。文章提供了一个具体的C++实现示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

树状数组求解

离线

模板

#include<iostream>
#include<algorithm>

using namespace std;
const int N=300005,M=200005;
struct Query{
	int l,r,id;
	bool operator<(const Query &a)const
	{
		return r==a.r?(l<a.l):(r<a.r);
	}
}q[M];

int n,sum[N],a[N],ans[N],vis[N];

int lowbit(int x)
{
	return x&(-x);
}

void change(int x,int p)
{
	while(x<=n){
		sum[x]+=p;
		x+=lowbit(x);
	}
}

int getsum(int x)
{
	int ans=0;
	while(x>0){
		ans+=sum[x];
		x-=lowbit(x);
	}
	return ans;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	  scanf("%d",&a[i]);
	int Q;
	scanf("%d",&Q);
	for(int i=0;i<Q;i++){
		scanf("%d%d",&q[i].l,&q[i].r);
		q[i].id=i;
	}
	sort(q,q+Q);
	int num=0;
	for(int i=1;i<=n&&num<Q;i++){
		if(vis[a[i]])
		  change(vis[a[i]],-1);
		change(i,1);
		vis[a[i]]=i;
		while(q[num].r<=i&&num<Q){
			ans[q[num].id]=getsum(q[num].r)-getsum(q[num].l-1);
			num++;
		}
	}
	for(int i=0;i<Q;i++)
	  printf("%d\n",ans[i]);
	return 0;
}