洛谷P2709 小B的询问基础莫队

最新推荐文章于 2024-07-09 01:32:36 发布

原创最新推荐文章于 2024-07-09 01:32:36 发布 · 108 阅读

1 ·

CC 4.0 BY-SA版权

莫队——基础莫队专栏收录该内容

7 篇文章

订阅专栏

本文详细解析洛谷P2709小B的询问题目，介绍如何使用莫队算法求解区间内所有数出现次数的平方和。提供了完整的AC代码实现，包括预处理和查询处理过程。

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

洛谷P2709 小B的询问

前言

我的csdn和博客园是同步的，欢迎来访danzh-博客园~

简明题意

求区间[L,R]中所有的数出现次数的平方和

思路

就是基础莫队的基础题了~下面是我的模板：

void add(int x)
{}
void remove(int x)
{}//这两个函数都是用来处理L,R指针移动时对答案的贡献,x是移走/来的点的id
for (int i = 1; i <= m; i++)
{
	while (l < query[i].l) remove(l++);
	while (l > query[i].l) add(--l);
	while (r < query[i].r) add(++r);
	while (r > query[i].r) remove(r--);
	
	ans[query[i].id] = ans;
}
//下面是预处理
int len = sqrt(n);
...
query[i].k = (query[i].l - 1) / len + 1;//k是指块的id
//就这么些~很简单~

注意事项

一开始令l=1，r=0

总结

AC代码

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;

const int maxn = 50000 + 10;

struct Query
{
	int l, r, id, k;
	bool operator < (const Query& a)const
	{
		if (k == a.k) return r < a.r;
		else return k < a.k;
	}
};

int n, m, k, a[maxn];
Query query[maxn];
int ans0[maxn];

int ans, cnt[maxn];
void add(int x)
{
	x = a[x];
	ans += 2 * cnt[x] + 1;
	cnt[x]++;
}
void remove(int x)
{
	x = a[x];
	ans -= 2 * cnt[x] - 1;
	cnt[x]--;
}

void solve()
{
	scanf("%d%d%d", &n, &m, &k);
	int len = (int)sqrt(n);
	for (int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	for (int i = 1; i <= m; i++)
		scanf("%d%d", &query[i].l, &query[i].r), query[i].id = i, query[i].k = (query[i].l - 1) / len + 1;
	sort(query + 1, query + 1 + m);

	int l = 1, r = 0;
	for (int i = 1; i <= m; i++)
	{
		while (l < query[i].l) remove(l++);
		while (l > query[i].l) add(--l);
		while (r < query[i].r) add(++r);
		while (r > query[i].r) remove(r--);
		
		ans0[query[i].id] = ans;
	}

	for (int i = 1; i <= m; i++)
		printf("%d\n", ans0[i]);
}

int main()
{
	solve();
	return 0;
}