P1494 [国家集训队] 小 Z 的袜子 Solution

Description

给定序列 $a=(a_1,a_2,\cdots ,a_n)$，有 $q$ 次查询，每次查询给定 $(l,r)$.
你需要求出 $\frac{2{\sum }_{i\le i<j\le r}[a_i=a_j]}{(r-l)(r-l+1)}$，以最简分数形式输出.

Limitations

$1\le n,m\le 5\times 10^4$
$1\le l\le r\le n$
$1\le a_i\le n$
$0.2\text{s},128\text{MB}$

Solution

这种不好直接维护的题，考虑上莫队.
我们设 $cn{t}_i$ 为当前区间内 $i$ 的出现次数.
那么，这 $cn{t}_i$ 个 $i$ 可以两两配成满足条件的对，其对分子的贡献显然为 ${\text{cnt}}_i\times ({\text{cnt}}_i-1)$.
在 add 和 del 时，先将原来 $a_x$ 的贡献减掉，更新 $\text{cnt}$ 后再加回去.
剩下的就是是模板，但有几个坑：

• 分子和分母会爆 int，要开 long long.
• 注意特判 $l=r$.
• 调块长，实测取 $B=\sqrt{n}$ 可以过.

Code

$2.41\text{KB},0.35\text{s},2.17\text{MB}\text{(in total, C++20 with O2)}$

#include <bits/stdc++.h>
using namespace std;

using i64 = long long;
using ui64 = unsigned long long;
using i128 = __int128;
using ui128 = unsigned __int128;
using f4 = float;
using f8 = double;
using f16 = long double;

template<class T>
bool chmax(T &a, const T &b){
	if(a < b){ a = b; return true; }
	return false;
}

template<class T>
bool chmin(T &a, const T &b){
	if(a > b){ a = b; return true; }
	return false;
}

namespace fastio {}   // Removed
using fastio::read;
using fastio::write;

struct Query { 
    int l, r, id;
    inline Query() {}
    inline Query(int _l, int _r, int _id) : l(_l), r(_r), id(_id) {}
}; 

signed main() {
	ios::sync_with_stdio(0);
	cin.tie(0), cout.tie(0);
	
	const int n = read<int>(), m = read<int>();
	const int B = sqrt(n);
	vector<int> c(n);
	for (int i = 0; i < n; i++) c[i] = read<int>(), c[i]--;
	
	i64 res = 0;
	vector<int> cnt(n);
	auto f = [&](int x) { return 1LL * x * (x - 1) / 2; };
	auto upd = [&](int x, int k) {
		res -= f(cnt[c[x]]);
		cnt[c[x]] += k;
		res += f(cnt[c[x]]);
	};
	
	vector<Query> qry(m);
	for (int i = 0, l, r; i < m; i++) {
		l = read<int>(), r = read<int>(), l--, r--;
		qry[i] = Query(l, r, i);
	}
	
	sort(qry.begin(), qry.end(), [&](const Query& a, const Query& b) {
		return (a.l / B == b.l / B) ? (a.r / B < b.r / B) : (a.l / B < b.l / B);
	});
	
	int l = 0, r = -1;
	vector<i64> num(m), den(m);
	for (auto& [ql, qr, id] : qry) {
		while (ql < l) upd(--l, 1);
        while (r < qr) upd(++r, 1);
        while (l < ql) upd(l++, -1);
        while (qr < r) upd(r--, -1);
        const i64 a = res, b = f(qr - ql + 1), g = gcd(a, b);
        if (a == 0) num[id] = 0, den[id] = 1;
        else num[id] = a / g, den[id] = b / g;
	}
	
	for (int i = 0; i < m; i++) {
		write(num[i]), putchar_unlocked('/');
		write(den[i]), putchar_unlocked('\n');
	}
	
	return 0;
}