Gym - 101606F-Flipping Coins 概率DP

题意：

给出n个硬币，初始时，n个硬币都是正面朝下的。现在你有k次机会将硬币投掷到空中，问你k次投掷后硬币正面朝上的个数的期望是多少

题解：

我们设DP[i][j] 表示 i 次 投掷后有j个硬币正面朝上的概率

最后的答案为∑(DP[k][i] * i)

转移方程为: DP[i+1][j] = DP[i][j] * 0.5              选择一个反面的投掷且投掷结果为反面

                        DP[i+1][j+1] = DP[i][j] * 0.5          选择一个反面的投掷且投掷结果为正面

需要注意的是

                        当j == n时有:

                        DP[i+1][n] = DP[i][n]*0.5                只能选一个正面的投掷，且投掷的结果为正面

                        DP[i+1][n-1] = DP[i][n] * 0.5      因为必须投掷k次，所以当已经全为正面的时候，

                                                                                         有1/2的概率使得一个正面的变成反面

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 500;
double dp[maxn][maxn];
double solve(int n,int k) {
    memset(dp,0,sizeof(dp));
    dp[0][0] = 1;
    for(int i=0;i<k;i++) {
        for(int j=0;j<n;j++) {
            dp[i+1][j] += dp[i][j] * 0.5;
            dp[i+1][j+1] += dp[i][j] * 0.5;
        }
        dp[i+1][n] += dp[i][n] * 0.5;
        dp[i+1][n-1] += dp[i][n] * 0.5;
    }
    double ans = 0;
    for(int i=1;i<=n;i++) ans += dp[k][i] * i;
    return ans;
}
int main()
{
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        printf("%.6f\n",solve(n,k));
    }
    return 0;
}