题目传送门
https://lydsy.com/JudgeOnline/problem.php?id=5016
题解
原式是这样的
\[\sum_{x = 0}^{\infty} get(l_1, r_1, x) \cdot get(l_2, r_2, x)\]
因为一次询问需要用到两个区间，本来按理说最擅长两个区间的查询的主席树，这里也没有办法建立。
然后分块或者莫队的话也无能为力。
于是我们考虑对询问本身做一些修改，使得一次询问只涉及两个参数。
\[\sum_{x = 0}^{\infty} get(l_1, r_1, x) \cdot get(l_2, r_2, x)\\= \sum_{x=0}^{\infty} get(1, r_1, x) \cdot get(1, r_2, x) - get(1, l1 - 1, x) \cdot get(1, r2, x) - get(1, l2 - 1, x) \cdot get(1, r1, x) + get(1, l1 - 1, x) \cdot get(1, l2 - 1, x)\]
这样，我们令 \(g(a, b)\) 表示 \(\sum\limits_{x = 0}^{\infty} get(1, a, x) \cdot get(1, b, x)\)，那么我们就把上面的一个询问分解成了 \(4\) 个询问。
然后 \(g(a, b)\) 可以对 \(a, b\) 进行莫队完成。
时间复杂度 \(O(n\sqrt{4m})\)。这样写复杂度似乎不是很标准，因为复杂度里面不应该有任何常数。
#include<bits/stdc++.h>
#define fec(i, x, y) (int i = head[x], y = g[i].to; i; i = g[i].ne, y = g[i].to)
#define dbg(...) fprintf(stderr, __VA_ARGS__)
#define File(x) freopen(#x".in", "r", stdin), freopen(#x".out", "w", stdout)
#define fi first
#define se second
#define pb push_back
template<typename A, typename B> inline char smax(A &a, const B &b) {return a < b ? a = b, 1 : 0;}
template<typename A, typename B> inline char smin(A &a, const B &b) {return b < a ? a = b, 1 : 0;}
typedef long long ll; typedef unsigned long long ull; typedef std::pair<int, int> pii;
template<typename I> inline void read(I &x) {
int f = 0, c;
while (!isdigit(c = getchar())) c == '-' ? f = 1 : 0;
x = c & 15;
while (isdigit(c = getchar())) x = (x << 1) + (x << 3) + (c & 15);
f ? x = -x : 0;
}
const int N = 50000 + 7;
#define bl(x) (((x) - 1) / blo + 1)
int n, m, blo, Q;
ll val;
int a[N], cl[N], cr[N];
ll ans[N];
struct Query {
int opt, l, r;
ll *ans;
inline Query() {}
inline Query(const int &opt, const int &l, const int &r, ll *ans) : opt(opt), l(l), r(r), ans(ans) {
if (l > r) std::swap(this->l, this->r);
}
inline bool operator < (const Query &b) const { return bl(l) != bl(b.l) ? l < b.l : r < b.r; }
} q[N << 2];
inline void addl(int x) {
val += cr[a[x]];
++cl[a[x]];
}
inline void addr(int x) {
val += cl[a[x]];
++cr[a[x]];
}
inline void dell(int x) {
val -= cr[a[x]];
--cl[a[x]];
}
inline void delr(int x) {
val -= cl[a[x]];
--cr[a[x]];
}
inline void work() {
blo = sqrt(Q);
std::sort(q + 1, q + Q + 1);
int l = 0, r = 0;
for (int i = 1; i <= Q; ++i) {
while (r < q[i].r) addr(++r);
while (l < q[i].l) addl(++l);
while (l > q[i].l) dell(l--);
while (r > q[i].r) delr(r--);
*q[i].ans += q[i].opt * val;
}
for (int i = 1; i <= m; ++i) printf("%lld\n", ans[i]);
}
inline void init() {
read(n);
for (int i = 1; i <= n; ++i) read(a[i]);
read(m);
for (int i = 1; i <= m; ++i) {
int l1, r1, l2, r2;
read(l1), read(r1), read(l2), read(r2);
q[++Q] = Query(1, r1, r2, ans + i);
if (l2 > 1) q[++Q] = Query(-1, l2 - 1, r1, ans + i);
if (l1 > 1) q[++Q] = Query(-1, l1 - 1, r2, ans + i);
if (l1 > 1 && l2 > 1) q[++Q] = Query(1, l1 - 1, l2 - 1, ans + i);
}
}
int main() {
#ifdef hzhkk
freopen("hkk.in", "r", stdin);
#endif
init();
work();
fclose(stdin), fclose(stdout);
return 0;
}