洛谷P2709 小B的询问 - 莫队

首先双关键字排序，然后扫一遍
考虑每次移动指针后答案的改变

#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;
}