看到长这样的题目,显然是莫队板子题。
但是不知道为什么很多人写的都是 \(2 \times cnt_x + 1\) 之类的?好像直接先减再加不就好了?公式都不用推。
注意指针顺序以及 long long
。
目前 CF 的机子上已经不需要用 %l64d
输出 long long
,直接 %lld
输出即可。
代码:
#include <bits/stdc++.h> using namespace std; typedef long long LL; const int MAXA = 1e6 + 10, MAXN = 2e5 + 10; int n, m, cnt[MAXA], a[MAXN], ys[MAXN], block; LL total, ans[MAXN]; struct node { int l, r, id; }q[MAXN]; int read() { int sum = 0, fh = 1; char ch = getchar(); while (ch < '0' || ch > '9') {if (ch == '-') fh = -1; ch = getchar();} while (ch >= '0' && ch <= '9') {sum = (sum << 3) + (sum << 1) + (ch ^ 48); ch = getchar();} return sum * fh; } bool cmp(const node &fir, const node &sec) { if (ys[fir.l] ^ ys[sec.l]) return ys[fir.l] < ys[sec.l]; if (ys[fir.l] & 1) return fir.r < sec.r; return fir.r > sec.r; } void add(int x) { total -= (LL)cnt[a[x]] * cnt[a[x]] * a[x]; cnt[a[x]]++; total += (LL)cnt[a[x]] * cnt[a[x]] * a[x]; } void del(int x) { total -= (LL)cnt[a[x]] * cnt[a[x]] * a[x]; cnt[a[x]]--; total += (LL)cnt[a[x]] * cnt[a[x]] * a[x]; } int main() { n = read(), m = read(); block = sqrt(n); for (int i = 1; i <= n; ++i) a[i] = read(); for (int i = 1; i <= n; ++i) ys[i] = (i - 1) / block + 1; for (int i = 1; i <= m; ++i) {q[i].l = read(), q[i].r = read(), q[i].id = i;} sort(q + 1, q + m + 1, cmp); int l = 1, r = 0; for (int i = 1; i <= m; ++i)//l--,r++,r--,l++ { while (l > q[i].l) add(--l); while (r < q[i].r) add(++r); while (r > q[i].r) del(r--); while (l < q[i].l) del(l++); ans[q[i].id] = total; } for (int i = 1; i <= m; ++i) printf("%lld\n", ans[i]); return 0; }