P1494 小Z的襪子

P1494 小Z的襪子

題目描述
小Z把這N只襪子從1到N編號，而後從L到R編號,小Z只在乎襪子的顏色。
求多個 (L,R)中有多大的機率抽到兩隻顏色相同的襪子。
數據中有 \(L=R\) 的狀況，請特判這種狀況，輸出\(0/1\)。
輸入格式
輸入文件第一行包含兩個正整數 N 和 M。N爲襪子的數量，M爲小Z所提的詢問的數量。接下來一行包含 N 個正整數 \(C_i\)
表示第i只襪子的顏色，相同的顏色用相同的數字表示。再接下來 M 行，每行兩個正整數 L, R 表示一個詢問。

輸出格式
包含 M 行，對於每一個詢問在一行中輸出分數 A/B 表示從該詢問的區間 [L,R] 中隨機抽出兩隻襪子顏色相同的機率。
若該機率爲 0 則輸出 0/1，不然輸出的 \(A/B\) 必須爲最簡分數。

題解
因爲過於懶惰，題解傳送門。反正是給本身看的

#include <cstdio>
#include <iostream>
#include <cmath> 
#include <algorithm>
#define orz cout<<"AK IOI"
#define int long long

using namespace std;
const int maxx = 50010;

inline int read()
{
	int x = 0, f = 1;
	char ch = getchar();
	while (ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
	while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
	return x * f;
}
int n, m, a[maxx];//a爲襪子的顏色 
int qn, l = 0, r = -1, ans, cs[maxx];//初始保證l r之間爲空 
int dafz[maxx], dafm[maxx];
struct node{
	int l, r, id, k;
}q[maxx];
bool cmp(node a, node b)
{
	if(a.k < b.k || (a.k == b.k && a.r < b.r)) return 1;
	return 0;
}
int c(int x)
{
	if(x < 2) return 0;
	return x * (x - 1) / 2;  
}
void delet(int x)
{
    int tp = --cs[a[x]];
    ans -= c(tp + 1);
    ans += c(tp);
}
void add(int x)
{
    int tp = ++cs[a[x]];
    ans -= c(tp - 1);
    ans += c(tp);
}
int gcd(int a,int b)
{
	if(b % a == 0) return a;
    return gcd(b, a%b);
}
signed main()
{
	//freopen("P1494_7.in","r",stdin);
	//freopen("P1494.out","w",stdout); 
	n = read(); m = read();
	for(int i = 1; i <= n; i++)
	a[i] = read();
	qn = sqrt(n + 0.5);//每塊的大小 和 一共有多少塊
	for(int i = 1; i <= m; i++)
	{
		q[i].l = read(); q[i].r = read();
		q[i].id = i;
		q[i].k = q[i].l / qn;//在第幾塊 
	}
	sort(q + 1, q + m + 1, cmp);//對查詢進行排序 
	for(int i = 1; i <= m; i++)
	{
		while(l < q[i].l)
		{
			delet(l);//由於當前的l值不包含在要查詢的區間內，先刪再減 
			l++;
		}
		while(l > q[i].l)
		{
		    l--;//由於當前的l值包含在要查詢的區間內，先加再加上當前 
			add(l);
		}
		while(r < q[i].r)
		{
		    r++;
			add(r);
		}
		while(r > q[i].r)
		{
		    delet(r);
			r--;
		}
		dafz[q[i].id] = ans;                 
		dafm[q[i].id] = c(r - l + 1);
	} 
	for(int i = 1; i <= m; i++)                 
	{
		if(dafz[i] == 0)
		{
			printf("0/1\n");
			continue;
		}
		int tp = gcd(dafz[i], dafm[i]);
		printf("%lld/%lld\n",dafz[i]/tp, dafm[i]/tp);
	}
	return 0;
}