洛谷 #2709. 小B的询问

最新推荐文章于 2022-04-01 22:11:16 发布

时间次元

最新推荐文章于 2022-04-01 22:11:16 发布

阅读量129

点赞数

CC 4.0 BY-SA版权

分类专栏：莫队

本文链接：https://blog.youkuaiyun.com/qq_41593522/article/details/84501986

莫队专栏收录该内容

3 篇文章

订阅专栏

本文深入解析了莫队算法的原理与应用，通过一道具体的题目，详细展示了如何使用莫队算法解决区间查询问题，包括算法的实现细节、调试过程及优化技巧。

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

题意

求l~r中每个数出现次数的平方和

题解

莫队

调试记录

平方差公式写错了

#include <cstdio>
#include <cmath>
#include <algorithm>
#define maxn 50005

using namespace std;

struct node{
	int l, r, id, ans;
}q[maxn];

int n, m, k, a[maxn], pos[maxn], cnt[maxn];

bool cmp(node const &a, node const &b){ return (pos[a.l] == pos[b.l]) ? (pos[a.l] & 1) ? a.r < b.r : a.r > b.r : a.l < b.l; }
bool idcmp(node const &a, node const &b){ return a.id < b.id; }

void solve(){
	int l = 1, r = 0, ans = 0;
	for (int i = 1; i <= m; i++){
		for (; r < q[i].r; r++) cnt[a[r + 1]]++, ans += (cnt[a[r + 1]] * 2 - 1);
		for (; r > q[i].r; r--) ans -= (cnt[a[r]] * 2 - 1), cnt[a[r]]--;
		for (; l < q[i].l; l++) ans -= (cnt[a[l]] * 2 - 1), cnt[a[l]]--;
		for (; l > q[i].l; l--) cnt[a[l - 1]]++, ans += (cnt[a[l - 1]] * 2 - 1);
		q[i].ans = ans;
	}
}

int main(){
	scanf("%d%d%d", &n, &m, &k);
	
	for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
	int block = sqrt(m);
	for (int i = 1; i <= m; i++) pos[i] = (i - 1) / block + 1;
	for (int i = 1; i <= m; i++) scanf("%d%d", &q[i].l, &q[i].r), q[i].id = i;
	
	sort(q + 1, q + m + 1, cmp);
	solve();
	sort(q + 1, q + m + 1, idcmp);
	
	for (int i = 1; i <= m; i++) printf("%d\n", q[i].ans);
	
	return 0;
}