luoguP1494小Z的袜子

题目描述

小Z的袜子

（因自己的gcd没有return而RE了一天的题 ）

作为一个生活散漫的人，小 Z 每天早上都要耗费很久从一堆五颜六色的袜子中找出一双来穿。终于有一天，小 Z 再也无法忍受这恼人的找袜子过程，于是他决定听天由命……

具体来说，小 Z 把这 N 只袜子从 1 到 N 编号，然后从编号 L 到 R (L 尽管小 Z 并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。

你的任务便是告诉小 Z，他有多大的概率抽到两只颜色相同的袜子。当然，小 Z 希望这个概率尽量高，所以他可能会询问多个(L,R)以方便自己选择。

然而数据中有L=R的情况，请特判这种情况，输出0/1。

输入格式

输入文件第一行包含两个正整数 N 和 M。N 为袜子的数量，M 为小 Z 所提的询问的数量。接下来一行包含 N 个正整数 $C_i$,其中 $C_i$表示第 ii 只袜子的颜色,相同的颜色用相同的数字表示。再接下来 M 行，每行两个正整数 L,R表示一个询问。

输出格式

包含 M 行，对于每个询问在一行中输出分数 A/B表示从该询问的区间 $[L,R]$中随机抽出两只袜子颜色相同的概率。若该概率为 0则输出 0/1,否则输出的A/B必须为最简分数。

思路

莫队经典假 板子(被卡了一天以为是毒瘤 ),其实并不需要每次求出A和B分别为多少,只用求出可行方案数和总方案数即可.
此时我们需要用数组$cnt[i]$来维护每次移动后区间包含每种袜子的个数,len表示区间长度(answer=可行方案数/总方案数).
我们可以发现:每次新增一只袜子x,可行方案数就增加$cnt[x]$,总方案数就增加$(len+1)\ast len/2-(len-1)\ast len/2$即$len$.
通过此结论,我们就可以把这个假 板子变成真板子了!

代码

#include<bits/stdc++.h>
#define in read()
#define N 100005
using namespace std;

int n,m,fz=0,fm=0,len;
int a[N],ans[N][2],cnt[N];
struct node{
int l,r,poi,pos;}q[N];

inline int in{
	int i=0;char ch;
	while(!isdigit(ch)){ch=getchar();}
	while(isdigit(ch)){i=(i<<3)+(i<<1)+(ch^48);ch=getchar();}
	return i;
}//必要的快读优化(在只有100ms时限的题中非常有用!)

inline bool cmp(const node &a,const node &b)//分块
{
	if(a.pos==b.pos)
	{
		return a.r<b.r;
	}
	return a.pos<b.pos;
}

inline int gcd(int a,int b)
{
	if(b==0)return a;
	return gcd(b,a%b);//一定要加return,我就是因为这个被卡了一天
}

inline void add(int x)//加袜子操作
{
	fz+=cnt[a[x]];
	++cnt[a[x]];
	fm+=len;
	++len;
}

inline void jian(int x)//除袜子操作
{
	--cnt[a[x]];
	fz-=cnt[a[x]];
	--len;
	fm-=len;
}

int main()
{
	n=in,m=in;
	int di=sqrt(n);
	for(int i=1;i<=n;i++)
	a[i]=in;
	for(int i=1;i<=m;i++)
	{
		q[i].l=in,q[i].r=in;
		q[i].poi=i,q[i].pos=(q[i].l+di-1)/di;
	}
	sort(q+1,q+m+1,cmp);
	int l=1,r=0,g;
	for(int i=1;i<=m;i++)//莫队
	{
		int ll=q[i].l,rr=q[i].r;
		if(ll==rr)
		{
			ans[q[i].poi][0]=0;
			ans[q[i].poi][1]=1;
			continue;
		}
		while(l<ll)jian(l++);
		while(r<rr)add(++r);
		while(l>ll)add(--l);
		while(r>rr)jian(r--);
		if(fz==0)
		{
			ans[q[i].poi][0]=0;
			ans[q[i].poi][1]=1;
		}
		else 
		{
			g=gcd(fz,fm);
			ans[q[i].poi][0]=fz/g;
			ans[q[i].poi][1]=fm/g;
		}
	}
	for(int i=1;i<=m;i++)
	printf("%d/%d\n",ans[i][0],ans[i][1]);
	return 0;
}