题目大意:有一个序列$s_i$(最大的数为$k$),有$m$个询问,询问$[l,r]$中$sum_{i=1}^{k}c_i^2$($c_i$表示数字$i$在$[l,r]$中的出现次数)。
题解:莫队
卡点:为什么我奇偶性优化会锅???(后记:发现了,用异或时,若($a.l==b.l&&a.r==b.r$时可能会出现$a<b=true&&b<a=true$)
C++ Code:
#include <cstdio> #include <algorithm> #define bsz 224 #define belong(i) ((i / bsz) + (i % bsz && 1)) #define maxn 50010 int n, m, k; int ans[maxn]; int s[maxn], cnt[maxn]; struct node { int l, r, num; inline bool operator < (const node &rhs) const {return (belong(l) == belong(rhs.l)) ? ((belong(l) & 1) ? r < rhs.r : r > rhs.r) : l < rhs.l;} } q[maxn]; int main() { scanf("%d%d%d", &n, &m, &k); 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].num = i; std::sort(q + 1, q + m + 1); int l = 1, r = 1, res = 1; cnt[s[1]]++; for (int i = 1; i <= m; i++) { while (l > q[i].l) res += cnt[s[--l]]++ << 1 | 1; while (r < q[i].r) res += cnt[s[++r]]++ << 1 | 1; while (l < q[i].l) res -= --cnt[s[l++]] << 1 | 1; while (r > q[i].r) res -= --cnt[s[r--]] << 1 | 1; ans[q[i].num] = res; } for (int i = 1; i <= m; i++) printf("%d ", ans[i]); return 0; }