模板——莫队

莫队ACM模板

题目描述
第一行，三个整数N、M、K。（M个询问）
第二行，N个整数，表示序列，（K>=每个数>=1）。
接下来的M行，每个询问给定区间[L，R]
求此区间sum(c(1)^2+ c(2)^2+···+ c(K)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L，R]中的个数。
输出：M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

const int maxn = 50010;//N的上限
int pos[maxn];//pos[i]：第i个数属于第几块
int a[maxn];//a[i]：第i个数
int c[maxn];//c[i]：数字i出现的个数
int ans[maxn], Ans;//ans[i]：第i个询问的答案
int n, m, k;//第一行的N，M，K

struct node {
	int l, r, id;//第id个询问的区间是[l,r]
}s[maxn];
bool cmp(node a, node b) {//排序依据
	if (pos[a.l] == pos[b.l]) {
		return a.r < b.r;
	}
	else {
		return pos[a.l] < pos[b.l];
	}
}
void add(int i) {//把第i个数添加到区间里
	Ans -= c[a[i]] * c[a[i]];
	c[a[i]]++;
	Ans += c[a[i]] * c[a[i]];
};
void del(int i) {//把第i个数从区间里删除
	Ans -= c[a[i]] * c[a[i]];
	c[a[i]]--;
	Ans += c[a[i]] * c[a[i]];
};
void reset() {//重置
	memset(a, 0, sizeof(a));
	memset(c, 0, sizeof(c));
	int block = (int)sqrt(n);
	for (int i = 1; i <= n; i++) {
		pos[i] = (i-1) / block+1;
	}
	for (int i = 1; i <= m; i++) {
		s[i].id = i;
	}
}
void md() {//更新ans数组
	int l = 1, r = 0;
	Ans = 0;
	sort(s + 1, s + m + 1, cmp);
	for (int i = 1; i <= m; i++) {
		while (l < s[i].l) {
			del(l);
			l++;
		}
		while (l > s[i].l) {
			l--;
			add(l);
		}
		while (r < s[i].r) {
			r++;
			add(r);
		}
		while (r > s[i].r) {
			del(r);
			r--;
		}
		ans[s[i].id] = Ans;
	}
}