HDU3874(莫队算法，模板）

原创于 2019-03-28 00:14:29 发布 · 256 阅读

0 ·

CC 4.0 BY-SA版权

莫队算法专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种解决区间问题的有效算法——莫队算法，并通过具体代码实现了解决区间内数字求和的问题，确保了重复数字仅被计算一次。该算法特别适用于临近点转移较为便捷的情形。

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

题意：给定一个数组，每次询问一个区间[l, r]，求区间内所有数字的和，出现多次的数字只加一次

解题思路：莫队算法就适用于这种区间问题，且临近点转移比较方便的情况。复杂度最大应该也有O(M *sqrt(n)),但总比暴力的O(MN）要好。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 50009
#define M 200009 
#define ll long long
using namespace std;
struct node
{
	int l,r,id;
}q[M];
ll ans;
int s[N],vis[1000008],be[N],bck;
ll res[M];
bool cmp(node a,node b)
{
	if(be[a.l]==be[b.l]) return a.r<b.r;
	return a.l<b.l;
}
void add(int pos)
{
	if(!vis[s[pos]])
	ans+=s[pos];
	vis[s[pos]]++;
}
void del(int pos)
{
	if(vis[s[pos]]==1)
	ans-=s[pos];
	vis[s[pos]]--;
}
int main()
{
	//freopen("t.txt","r",stdin);
	int T,l,r,n,m;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		memset(vis,0,sizeof(vis));
		bck=sqrt(n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&s[i]);
			be[i]=i/bck+1;
		}
		scanf("%d",&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d",&q[i].l,&q[i].r);
			q[i].id=i;
		}
		sort(q+1,q+1+m,cmp);
		ans=0;
		int l=1,r=0;
		for(int i=1;i<=m;i++)
		{
			while(l>q[i].l) add(l-1),l--;
			while(r<q[i].r) add(r+1),r++;
			while(l<q[i].l) del(l),l++;
			while(r>q[i].r) del(r),r--;
			res[q[i].id]=ans;
		}
		for(int i=1;i<=m;i++)
		printf("%lld\n",res[i]);		
	 } 
	return 0;
}