[CF457D] Bingo!

原创于 2020-04-11 13:56:28 发布 · 289 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

组合数同时被 2 个专栏收录

2 篇文章

订阅专栏

动态规划# 概率dp

1 篇文章

订阅专栏

本文深入探讨了一道结合概率动态规划与组合数学的问题，详细解释了如何通过计算特定状态下行列数的组合来求解得分期望值。文章提供了完整的解析过程，包括状态贡献的计算、概率的定义以及组合数的高效计算方法。

题意

洛谷
在这里插入图片描述

思路

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

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

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

设 $P_{x,y}$ 为选出 $x$ 行 $y$ 列被涂黑的概率，则状态 $S^{'}$ 选出 $x$ 行 $y$ 列的方案数为 $P_{x,y}*C_{n}^{x}C_{n}^{y}$ ，那么总的方案数就为 $\sum_{x=0}^{n}\sum_{y=0}^{n}P_{x,y}*C_{n}^{x}C_{n}^{y}$

下面考虑如何计算 $P_{x,y}$ 。 $m$ 中选 $k$ ，总共有 $C_{m}^{k}$ 种方案。容易知道，设 $N u m = x n + y n - x y$ ，就是这 $x$ 行 $y$ 列的总格子数，因为要被涂黑，所以这 $N u m$ 个数一定包含在选取的 $k$ 个数中。而其他的 $k - N u m$ 个数则随意选取，方案数为 $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))$ ，然而这道题是没有模数的，意味着阶乘算组合数时会爆炸，我们就需要一些方法计算组合数。

记 $l g [i] = l o g (i!)$ (~~这是一些同学最喜欢的数组~~)，得到 $l g [i] = l o g (i!) = l o g (i * (i - 1)!) = l g [i - 1] + l o g (i)$ ，这里大家学多项式的时候应该都知道了。我们知道， $C_{n}^{m}=JC[n]/JC[m]/JC[n-m]$ ，通过 $l g$ 数组，我们可以将阶乘的乘除转换为指数的加减，由此可以得到: $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;
}