欢迎访问 My Luogu Space。
【题目大意】
一段序列,每次询问 \([l, r]\) 范围内的每个数字的出现次数的平方和。
【题解】
莫队。
非常模板的莫队题。
推出 \((n+1)^2-n^2=2n+1\)(n为数字的出现次数);
意思是一个数字的出现次数多一次会对答案造成多少贡献。
分块分为 \(\sqrt{n}\) 个。
【代码】
// output format !!
// long long !!
#include <bits/stdc++.h>
#define H puts("HYX")
typedef long long LL;
const int INF = 0x3f3f3f3f;
const int MAXN = 50000+10;
int bol;
struct Q{
int l, r, id;
bool operator<(Q a)const{
return (r/bol==a.r/bol) ? l<a.l : r<a.r;
}
}q[MAXN];
int n, m, k;
int cnt[MAXN], s[MAXN];
LL ans, Ans[MAXN];
int main(){
scanf("%d%d%d", &n, &m, &k);
bol = sqrt(n);
for(int i=1; i<=n; ++i) scanf("%d", s+i);
for(int i=1; i<=m; ++i) scanf("%d%d", &q[i].l, &q[i].r), q[i].id = i;
std::sort(q+1, q+m+1);
int L = 1, R = 0;
for(int i=1; i<=m; ++i){
while(R-1 >= q[i].r) ans -= 1ll*cnt[s[R]]*2-1, --cnt[s[R--]];
while(R+1 <= q[i].r) ans += 1ll*cnt[s[++R]]*2+1, ++cnt[s[R]];
while(L+1 <= q[i].l) ans -= 1ll*cnt[s[L]]*2-1, --cnt[s[L++]];
while(L-1 >= q[i].l) ans += 1ll*cnt[s[--L]]*2+1, ++cnt[s[L]];
Ans[q[i].id] = ans;
}
for(int i=1; i<=m; ++i) printf("%lld\n", Ans[i]);
return 0;
}