[BZOJ3781][小B的询问][莫队]

题目：

有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。

思路：

裸的莫队吧。。

直接开数组 C 暴力维护每个数出现的次数，区间转移的时候如果假如加入一个数k，维护平方和就是：

ans=ans−C[k]2+(++C[k])2

反之减去一个数就是：

ans=ans−C[k]2+(−−C[k])2

代码：

#include <bits/stdc++.h> using namespace std; const int Maxn = 50005; typedef long long ll; inline ll sqr(const ll &x) { return x * x; } inline char get(void) { static char buf[1000000], *p1 = buf, *p2 = buf; if (p1 == p2) { p2 = (p1 = buf) + fread(buf, 1, 1000000, stdin); if (p1 == p2) return EOF; } return *p1++; } inline void read(int &x) { x = 0; static char c; for (; !(c >= '0' && c <= '9'); c = get()); for (; c >= '0' && c <= '9'; x = x * 10 + c - '0', c = get()); } int belong[Maxn]; ll ans[Maxn]; struct Cmd { int l, r, id; friend bool operator < (const Cmd &a, const Cmd &b) { if (belong[a.l] == belong[b.l]) return a.r < b.r; else return belong[a.l] < belong[b.l]; } } cmd[Maxn]; int n, m, k, c[Maxn], sum[Maxn]; inline void upd(ll &now, int p, int v) { now -= sqr(sum[c[p]]); sum[c[p]] += v; now += sqr(sum[c[p]]); } inline void solve(void) { int L = 1, R = 0; ll now = 0, g; for (int i = 1; i <= m; i++) { for (; L < cmd[i].l; L++) upd(now, L, -1); for (; R > cmd[i].r; R--) upd(now, R, -1); for (; L > cmd[i].l; L--) upd(now, L - 1, 1); for (; R < cmd[i].r; R++) upd(now, R + 1, 1); ans[cmd[i].id] = now; } } int main(void) { //freopen("in.txt", "r", stdin); read(n), read(m), read(k); int s = sqrt(n); for (int i = 1; i <= n; i++) { read(c[i]); belong[i] = (i - 1) / s + 1; } for (int i = 1; i <= m; i++) { read(cmd[i].l), read(cmd[i].r); cmd[i].id = i; } sort(cmd + 1, cmd + m + 1); solve(); for (int i = 1; i <= m; i++) { printf("%lld\n", ans[i]); } return 0; }

完。

By g1n0st