【莫队】luogu_2709 小B的询问

题意

$\sum {c_i}^2$的值,其中$i$的值从$1$到$K$，其中$c_i$表示数字$i$在$[L..R]$中的重复次数。

思路

莫队修改一下暴力中的两个函数即可，注意$l$从$1$开始，否则会多无用计算。

代码

#include<cmath>
#include<cstdio>
#include<algorithm>

struct node {
	int l, r, pos;
}que[500001];
int n, m, k, block, ans;
int a[500001], cnt[1000001], res[500001];

bool operator <(const node &x, const node &y) {
	return (x.l / block) ^ (y.l / block) ? x.l < y.l : (((x.l / block) & 1) ? x.r < y.r : x.r > y.r);
}

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

void del(int x) {
	ans -= 2 * cnt[a[x]] - 1;
	cnt[a[x]]--;
}

int main() {
	scanf("%d %d %d", &n, &m, &k);
	for (int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	block = n / sqrt(m * 2 / 3);
	for (int i = 1; i <= m; i++)
		scanf("%d %d", &que[i].l, &que[i].r), que[i].pos = i;
	std::sort(que + 1, que + m + 1);
	int l = 1, r = 0;
	for (int i = 1; i <= m; i++) {
		int ql = que[i].l, qr = que[i].r;
        while (l < ql) del(l++);
        while (l > ql) add(--l);
        while (r < qr) add(++r);
        while (r > qr) del(r--);
        res[que[i].pos] = ans;
	}
	for (int i = 1; i <= m; i++)
		printf("%d\n", res[i]);
}