洛谷P2709 小B的询问 - 莫队

按根下n分个块，对询问区间双关键字排序，若不在一个块里按左端点排序，若在按右端点排序，然后扫一遍统计答案
考虑每次移动指针后答案的改变
分为四种情况 分别是l, r指针在查询区间的l和r的左边还是右边，都讨论一下
每种情况该怎么做在纸上画画就明白了。。。没必要特意记

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
using namespace std;
#define debug(x) cerr << #x << "=" << x << endl;
const int MAXN = 50000 + 10;
int c[MAXN], be[MAXN],n,m,k,f[MAXN],ans;
struct mo {
	int l,r,val,id;
}a[MAXN];
bool cmpmo(mo a, mo b) {
	return be[a.l] == be[b.l] ? a.r < b.r : a.l < b.l;
}
bool cmpid(mo a, mo b) {
	return a.id < b.id;
}
void revise(int k, int op) {
	ans -= c[k]*c[k];
	c[k] += op;
	ans += c[k]*c[k];
}
int main() {
    cin >> n >> m >> k;
    int unit = sqrt(n);
    for(int i=1; i<=n; i++) {
    	be[i] = i/unit + 1;
    	scanf("%d", &f[i]);
    }
    for(int i=1; i<=m; i++) {
    	scanf("%d %d", &a[i].l, &a[i].r);
    	a[i].id = i;
    }
    sort(a+1, a+m+1, cmpmo);
    int l=1, r=0;
    for(int i=1; i<=m; i++) {
    	while(l < a[i].l) revise(f[l], -1), l++;
    	while(l > a[i].l) revise(f[l-1], 1), l--;//需要ai.l的贡献
    	while(r < a[i].r) revise(f[r+1], 1), r++;//需要ai.r的贡献
    	while(r > a[i].r) revise(f[r], -1), r--;
    	a[i].val = ans;	
    }
    sort(a+1, a+m+1, cmpid);
    for(int i=1; i<=m; i++) {
    	printf("%d\n", a[i].val);
    }
    return 0;
}