P1494 [国家集训队]小Z的袜子（普通莫队）

题目

传送门
题目描述
作为一个生活散漫的人，小Z每天早上都要耗费很久从一堆五颜六色的袜子中找出一双来穿。终于有一天，小Z再也无法忍受这恼人的找袜子过程，于是他决定听天由命……
具体来说，小Z把这N只袜子从1到N编号，然后从编号L到R(L 尽管小Z并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。
你的任务便是告诉小Z，他有多大的概率抽到两只颜色相同的袜子。当然，小Z希望这个概率尽量高，所以他可能会询问多个(L,R)以方便自己选择。
然而数据中有L=R的情况，请特判这种情况，输出0/1。
输入输出格式
输入格式：
输入文件第一行包含两个正整数N和M。N为袜子的数量，M为小Z所提的询问的数量。接下来一行包含N个正整数Ci，其中Ci表示第i只袜子的颜色，相同的颜色用相同的数字表示。再接下来M行，每行两个正整数L，R表示一个询问。
输出格式：
包含M行，对于每个询问在一行中输出分数A/B表示从该询问的区间[L,R]中随机抽出两只袜子颜色相同的概率。若该概率为0则输出0/1，否则输出的A/B必须为最简分数。（详见样例）
输入输出样例
输入样例#1：
6 4
1 2 3 3 3 2
2 6
1 3
3 5
1 6
输出样例#1：
2/5
0/1
1/1
4/15
说明
30%的数据中 N,M ≤ 5000；
60%的数据中 N,M ≤ 25000；
100%的数据中 N,M ≤ 50000，1 ≤ L < R ≤ N，Ci ≤ N。

题解

• 设我们要求的区间为$[L，R]$，区间长度为$len$，区间内每双袜子的数量为$a，b，\cdots ，n$，那么我们所要求的概率为$$\frac{C_a^2+C_b^2+\cdots +C_n^2}{C_{len}^2}$$
• 化简得$$\frac{a\ast (a-1)+b\ast (b-1)+\cdots +n\ast (n-1)}{len\ast (len-1)}$$
• 莫队离线维护即可

code

#include <bits/stdc++.h> 
using namespace std; 
const int maxn = 5e4 + 100; 
typedef long long LL; 

template <typename T>
inline void read(T &s) {
	s = 0; 
	T w = 1, ch = getchar(); 
	while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }
	while (isdigit(ch)) { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); }
	s *= w; 
}

LL ans;
LL f[maxn];  
int n, m; 
int a[maxn], pos[maxn]; 
struct query { 
	int l, r, id; 
	LL z, y; 
} q[maxn]; 

inline bool cmp(query A, query B) {
	return pos[A.l] == pos[B.l] ? A.r < B.r : A.l < B.l; 
}

inline bool cmp1(query A, query B) {
	return A.id < B.id; 
}

inline LL gcd(LL A, LL B) {
	return B == 0 ? A : gcd(B, A % B); 
}

inline void add(int x) {
	LL p = f[x]; 
	f[x]++; 
	LL q = f[x]; 
	ans -= p * (p - 1) / 2; 
	ans += q * (q - 1) / 2; 
}

inline void del(int x) {
	LL p = f[x]; 
	f[x]--; 
	LL q = f[x]; 
	ans -= p * (p - 1) / 2; 
	ans += q * (q - 1) / 2; 
}

int main() {
	read(n), read(m); 
	int blo = n / sqrt(m); 
	for (int i = 1; i <= n; ++i) {
		read(a[i]); 
		pos[i] = (i - 1) / blo + 1;
	}
	for (int i = 1; i <= m; ++i) {
		read(q[i].l), read(q[i].r); 
		q[i].id = i; 
	}
	sort(q + 1, q + m + 1, cmp); 
	
	for (int i = q[1].l; i <= q[1].r; ++i) add(a[i]); 
	q[1].z = ans;
	q[1].y = (q[1].r - q[1].l + 1) * (q[1].r - q[1].l) / 2; 
	if (q[1].z == 0) {
		q[1].y = 1; 
	}
	else {
		LL g = gcd(q[1].z, q[1].y); 
		q[1].z /= g; 
		q[1].y /= g; 	
	}
	
	int l = q[1].l, r = q[1].r; 
	for (int i = 1; i <= m; ++i) {
		int ql = q[i].l, qr = q[i].r; 
		while (l < ql) del(a[l++]); 
		while (l > ql) add(a[--l]); 
		while (r < qr) add(a[++r]);
		while (r > qr) del(a[r--]); 
		
		q[i].z = ans; 
		LL s = r - l + 1; 
		s = s * (s - 1); 
		q[i].y = s / 2; 
		if (q[i].z == 0) {
			q[i].y = 1; 
		}
		else {
			LL g = gcd(q[i].z, q[i].y); 
			q[i].z /= g; 
			q[i].y /= g; 
		}
	}
	
	sort(q + 1, q + m + 1, cmp1); 
	for (int i = 1; i <= m; ++i) 
		printf("%lld/%lld\n", q[i].z, q[i].y); 
	return 0; 
}