洛谷 P2709 小B的询问（莫队 模板）

小B的询问

题目描述

小B 有一个长为 $n$ 的整数序列 $a$，值域为 $[1,k]$。
他一共有 $m$ 个询问，每个询问给定一个区间 $[l,r]$，求：
$$\sum _{i=1}^k{c}_i^2$$

其中 $c_i$ 表示数字 $i$ 在 $[l,r]$ 中的出现次数。
小B请你帮助他回答询问。

输入格式

第一行三个整数 $n,m,k$。

第二行 $n$ 个整数，表示 小B 的序列。

接下来的 $m$ 行，每行两个整数 $l,r$。

输出格式

输出 $m$ 行，每行一个整数，对应一个询问的答案。

样例 #1

样例输入 #1

6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6

样例输出 #1

6
9
5
2

提示

【数据范围】
对于 $100\%$ 的数据，$1\le n,m,k\le 5\times 10^4$。

1、 莫队， 最原本的原型是， 有一个数列 a{n}（n个数）， 有m次查询， [l, r]
	每次查询输出 区间[l, r] 上所有数的和。
2、  Add() 和 Sub() 函数， 假如当前区间是  [l, r] , 和是 res， 那么区间
	[l - 1, r] 的值，就是 res += a[l - l]
	[l + 1, r] 的值， res -= a[l]
	[l, r + 1] 的值， res += a[r + 1]
	[l, r - 1] 的值， res -= a[r]
	
3 、 如果从 一个区间 [x1, y1] 的和 通过  Add() 和 Sub() 函数 来得到下一个区间 
	[x2, y2] 的时候，如果 [x1, y1] 和 [x2, y2] 差距很大，那么操作的复杂度很高。
	比如 区间序列 [1, 2], [n - 1, n] ,[1, 2], [n - 1, n], [1, 2], [n - 1, n] ...
	m次查询复杂度就是 O(m * n)
4 、莫队 ，分块， 排序思想
	把下标 [1 ~ n] 分块， 每一块大小 sqrt(n)， 一共 sqrt(n)。
	用结构体数组， 记录每一查询，先按区间左端点 l 所在的块来排序。
	（如果所在块相等）， 则按区间右端点 r 排序。
	struct Query
	{
		int l, r;	// 
		int idx;	//第几次查询
	}que[N];
	return pos[x.l] == pos[y.l] ? x.r < y.r : pos[x.l] < pos[y.l];
	
5、 分块排序后，本来凌乱的区间查询，现在就按照一定顺序， 从小到大排序了。
	也就是相邻的区间，它们的差距就小了很多，  调用Add() 和 Sub() 函数计算了，
	步骤就少了。

#include <bits/stdc++.h>
using namespace std;
const int N = 5e4 + 10;
int a[N];
int pos[N];		//分块
int cnt[N];		// cnt[i] 表示 i出现的次数
long long ans[N], res;
int n, m, k;

struct Query
{
	int l, r;	// 
	int idx;	//第几次查询
}que[N];

void Add(int x)
{
	cnt[a[x]]++;
	res += cnt[a[x]] * 2 - 1;
	// 值增加了 cnt[a[x]] * cnt[a[x]] - (cnt[a[x]] - 1) * (cnt[a[x]] - 1)
}

void Sub(int x)
{
	cnt[a[x]]--;
	res -= cnt[a[x]] * 2 + 1;
	//值减少了 (cnt[a[x]] + 1) * (cnt[a[x]] + 1) - cnt[a[x]] * cnt[a[x]]
}

int main()
{
	scanf("%d%d%d", &n, &m, &k);
	int siz = sqrt(n);
	for(int i = 1; i <= n; ++i)
	{
		scanf("%d", &a[i]);
		pos[i] = i / siz;
	}
	for(int i = 0; i < m; ++i)
	{
		scanf("%d%d", &que[i].l, &que[i].r);
		que[i].idx = i;
	}
	sort(que, que + m, [](Query x, Query y){
			return pos[x.l] == pos[y.l] ? x.r < y.r : pos[x.l] < pos[y.l];
			});

	int l = 1, r = 0;	// 维护当前的区间
	for(int i = 0; i < m; ++i)
	{
		while(que[i].l < l)	Add(--l);
		while(que[i].r > r)	Add(++r);
		while(que[i].l > l)	Sub(l++);
		while(que[i].r < r) Sub(r--);
		ans[que[i].idx] = res;
	}
	for(int i = 0; i < m; ++i)
		printf("%lld\n", ans[i]);

	return 0;
}