ACM-ICPC 2017 Asia Urumqi A.Coins解题报告

ACM-ICPC 2017 Asia Urumqi A.Coins

题目大意

有$n$枚硬币反面朝上，有$m$次操作，每次需要投掷$p$枚硬币，问你最后正面朝上的硬币个数的期望是多少

解题思路

$dp[i][j]$表示第$i$次投掷有$j$枚硬币正面朝上的概率，$k$表示第$i$次投掷$p$枚硬币后有$k$枚硬币正面朝上的概率，那么其转移方程即为：
$$\{\begin{array}{l}dp[i][j+k]=\sum dp[i-1][j]\times C_p^k\times 2^{-p}j\le n-p\\ dp[i][j-(p-(n-j))+k]=\sum dp[i-1][j]\times C_p^k\times 2^{-p}j>n-p\end{array}$$
其中第一个式子表示当前反面朝上的硬币数目多于$p$枚，那么我们只要从剩余的$n-j$枚硬币中选择$p$枚反面朝上的硬币进行投掷，最后成功将$k$枚硬币变成正面朝上的概率，第二个式子表示当期反面朝上的硬币数目少于$p$枚，那么我们就需要选择一些已经正面朝上的硬币进行投掷，其数目为$p-(n-j)$，同样，$k$表示投掷$p$枚硬币后正面朝上的硬币个数

则有期望$E={\sum }_{i=1}^{n}dp[m][i]\times i$

预处理出$100$以内的$C_k^p$和$2^{-p}$后根据公式计算即可

注：$C$数组开$longlong$会爆掉，需要使用$double$类型存储

AC代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 110;
double P[maxn];
double C[maxn][maxn];
double dp[maxn][maxn];
void pre() {
	P[0] = C[0][0] = 1.0;
	for (int i = 1; i < maxn; ++i) {
		C[i][0] = 1;
		for (int j = 1; j <= i; ++j) C[i][j] = C[i - 1][j - 1] + C[i - 1][j];
	}
	for (int i = 1; i < maxn; ++i) P[i] = P[i - 1] / 2.0;
}
int main() {
	pre();
	int T;
	scanf("%d", &T);
	while (T--) {
		int n, m, p;
		scanf("%d%d%d", &n, &m, &p);
		memset(dp, 0, sizeof(dp));
		dp[0][0] = 1;
		for (int i = 1; i <= m; ++i) {
			for (int j = 0; j <= n; ++j) {
				for (int k = 0; k <= p; ++k) {
					if (n - j >= p) dp[i][j + k] += dp[i - 1][j] * C[p][k] *  P[p];
					else dp[i][j - (p - (n - j)) + k] += dp[i - 1][j] * C[p][k] * P[p];
				}
			}
		}
		double res = 0;
		for (int i = 1; i <= n; ++i) res += dp[m][i] * i;
		printf("%.3f\n", res);
	}
	return 0;
}