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

最新推荐文章于 2025-09-08 21:57:50 发布

weixin_30673715

最新推荐文章于 2025-09-08 21:57:50 发布

阅读量50

点赞数

CC 4.0 BY-SA版权

文章标签： c/c++

原文链接：http://www.cnblogs.com/Mrsrz/p/8665773.html

本文介绍了一种解决洛谷P1494/BZOJ2038问题的方法，该问题要求计算从指定区间内随机选取两只颜色相同的袜子的概率。文章详细解释了解题思路，并给出了具体的C++实现代码。

题目：洛谷P1494、BZOJ2038。

题目大意：给你一列袜子的颜色，每次要你求从区间$[L,R]$内随机选两个袜子颜色相同的概率。
解题思路：
首先，对于某个特定区间$[L,R]$，它的概率是$\frac{\sum\limits_{i=1}^{n}\frac{C_i\times (C_i-1)}{2}}{\frac{(R-L+1)\times(R-L)}{2}}$，其中$C_i$表示在$[L,R]$内，颜色$i$的出现次数。
然后，$R-L+1$和$R-L$是可以直接求的。
重点在于求上面那个玩意儿。我们发现，如果$C_i$增加$1$，那么答案将会增加原来的$C_i$（减$1$同理）。
所以我们可以在知道$[L,R]$的答案的情况下，用$O(1)$的时间得到$[L\pm 1,R]$或$[L,R\pm 1]$的答案。
然后莫队即可。

C++ Code：

#include<bits/stdc++.h>
#define belong(i) ((i-1)/sz+1)
#define N 50005
#define ll long long
int n,sz,m,a[N],cnt[N]={0},p[N];
ll ans[N];
inline int readint(){
	int c=getchar();
	for(;!isdigit(c);c=getchar());
	int d=0;
	for(;isdigit(c);c=getchar())
	d=(d<<3)+(d<<1)+(c^'0');
	return d;
}
struct Q{
	int l,r,num;
	bool operator <(const Q& rhs)const{
		if(belong(l)!=belong(rhs.l))return l<rhs.l;
		return r<rhs.r;
	}
}q[N];
int main(){
	n=readint(),m=readint();
	sz=int(sqrt(n)+1e-5);
	for(int i=1;i<=n;++i)a[i]=readint();
	for(int i=1;i<=m;++i)q[i].l=readint(),q[i].r=readint(),q[i].num=i;
	std::sort(q+1,q+m+1);
	int l=1,r=1;
	ll now=0;
	++cnt[a[1]];
	for(int i=1;i<=m;++i){
		while(r<q[i].r)now+=cnt[a[++r]]++;
		while(l>q[i].l)now+=cnt[a[--l]]++;
		while(r>q[i].r)now-=--cnt[a[r--]];
		while(l<q[i].l)now-=--cnt[a[l++]];
		p[q[i].num]=r-l+1;
		ans[q[i].num]=now;
	}
	for(int i=1;i<=m;++i){
		if(ans[i]==0){
			putchar('0'),putchar('/'),putchar('1'),putchar('\n');
			continue;
		}
		int f=p[i];
		ll x=std::__gcd(ans[i],(ll)f*(f-1)>>1);
		printf("%lld/%lld\n",ans[i]/x,(ll)f*(f-1)/x>>1);
	}
	return 0;
}

转载于:https://www.cnblogs.com/Mrsrz/p/8665773.html