[CF457D] Bingo!

题意

洛谷

思路

与其说是一道概率dp，不如说是一道组合数学。

我们考虑计算一个状态$S$的贡献，它被涂黑的行列数记为$f(S)=x+y$，则对分数的贡献为$2^{f(S)}=2^{x+y}=2^x\ast 2^y$，所以$S$状态的分数贡献就是$x$行$y$列的子集数量。

设$S$状态的集合为$M$，则$S$的得分期望值为
$\frac{1}{|M|}{\sum }_{S^{\prime }\in S}{2}^{f(S^{\prime })}=\frac{1}{|M|}{\sum }_{S^{\prime }\in S}{\sum }_{i=0}^n{\sum }_{j=0}^n$(从$S^{\prime }$被完全标记的行列中选取$i$行$j$列的方案数)

设$P_{x,y}$为选出$x$行$y$列被涂黑的概率，则状态$S^{\prime }$选出$x$行$y$列的方案数为$P_{x,y}\ast C_n^x{C}_n^y$，那么总的方案数就为${\sum }_{x=0}^n{\sum }_{y=0}^n{P}_{x,y}\ast C_n^x{C}_n^y$

下面考虑如何计算$P_{x,y}$。$m$中选$k$，总共有$C_m^k$种方案。容易知道，设$Num=xn+yn-xy$，就是这$x$行$y$列的总格子数，因为要被涂黑，所以这$Num$个数一定包含在选取的$k$个数中。而其他的$k-Num$个数则随意选取，方案数为$C_{m-Num}^{k-Num}$，所以$P_{x,y}=\frac{C_{m-Num}^{k-Num}}{C_m^k}$。然而会发现，剩下的涂了之后会导致行、列数大于$x$或$y$，我们只需要改变$P_{x,y}$的含义为至少选出$x$行$y$列被涂黑的概率即可。

至此为止，公式就推出来了，可以套公式直接算，预处理组合数的话复杂度为$O(Max(n^2,m))$，然而这道题是没有模数的，意味着阶乘算组合数时会爆炸，我们就需要一些方法计算组合数。

记$lg[i]=log(i!)$(这是一些同学最喜欢的数组)，得到$lg[i]=log(i!)=log(i\ast (i-1)!)=lg[i-1]+log(i)$，这里大家学多项式的时候应该都知道了。我们知道，$C_n^m=JC[n]/JC[m]/JC[n-m]$，通过$lg$数组，我们可以将阶乘的乘除转换为指数的加减，由此可以得到:$C_n^m=exp(lg[n]-lg[m]-lg[n-m])$。这道题数据太大了，要使用long double,输出时转为double即可。

Code:

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#define MAXN 100005
#define Mod 1000000007
#define LL long long
#define DB double
#define LD long double
#define Int register int
using namespace std;
inline void read(LL &x)
{
	x = 0;
	LL f = 1;
	char s = getchar();
	while (s < '0' || s > '9')
	{
		if (s == '-')	
			f = -1;
		s = getchar();
	}
	while (s >= '0' && s <= '9')
	{
		x = (x << 3) + (x << 1) + (s ^ 48);
		s = getchar();
	}
	x *= f;
}
inline LL QuickPow(LL x,LL y)
{
	LL Res = 1, Temp = x;
	while ( y )
	{
		if (y & 1)
			Res = Res * Temp % Mod;
		Temp = Temp * Temp % Mod;
		y /= 2;
	}
	return Res;
}
LD lg[MAXN], Ans;
inline LD C(LL n,LL m)
{
	return lg[n] - lg[m] - lg[n - m];
}
int main()
{
	LL n, m, k;
	read( n ); read( m ); read( k );
	for (Int i = 2; i <= m; ++ i)
		lg[i] = lg[i - 1] + log( i );
	for (Int r = 0; r <= n; ++ r)
		for (Int c = 0; c <= n; ++ c)
		{
			LL Num = (r + c) * n - r * c;
			if (Num <= k)
				Ans += exp(C(n, r) + C(n, c) + C(m - Num, k - Num) - C(m, k));
		}
	if (Ans < 1e99)
		printf("%.10lf", (DB)Ans);
	else printf("1e99");
	return 0;
}