题目链接:
题目大意:
有n枚硬币,一开始全部正面朝下,对这n枚硬币进行m次操作,每次操作可以将n枚硬币中的任意k枚硬币抛起来,选择的策略是让最终尽可能多的硬币正面朝上,一枚硬币抛起来落地后正面朝上和反面朝上的概率均为1/2。问m次操作之后正面朝上的硬币的期望数量.
题解:
先求概率,概率*总数------>期望
dp[i][j] 是抛第i次有j枚硬币向上的概率
①既然策略是让最终尽可能多的硬币正面朝上,那么在当前正面朝下的硬币个数大于k的时候,肯定是在正面朝下的硬币中选择k个,本题因为朝上朝下的概率都是1/2,所以k个硬币抛下来,不管是有几个正面朝上,几个正面朝下,概率都是1/2^k
②如果当前朝下的硬币个数小于k个,那就要从朝上的里面再选一些
设当前有j个硬币是朝上的,那么就有n-j个硬币是朝下的,n-j<k,那么需要在朝上的硬币里面选k-(n-j)个,这样最后正面朝上的个数是本来正面就朝上的 j-(k-(n-j))+抛了之后朝上的h
=j-k+(n-j)+h=n-k+h
注意这里的组合数c数组要开double,会爆long long
#include<bits/stdc++.h>
#include<cstring>
#define ll long long
using namespace std;
double dp[110][110];
double deno[110];
double c[110][110];
void com(int n, int m)//m是没有用到的
{
memset(c, 0, sizeof(c));
c[0][0] = 1;
for(int i = 0; i <= n; i++)
{
c[i][0] = c[i][i] = 1;
for(int j = 1; j < i; j++)
{
c[i][j] = (c[i-1][j-1] + c[i-1][j]);
}
}
}
inline void init()
{
com(110,110);
deno[1]=0.5;
for(int i=2;i<=110;++i)
deno[i]=deno[i-1]/2;
//deno[i]就是1/2^i
}
int main()
{
//freopen("input.txt","r",stdin);
init();
int T;
cin>>T;
int n,m,k;
while(T--)
{
cin>>n>>m>>k;
memset(dp,0,sizeof(dp));
dp[0][0]=1.0;
for(int i=0;i<m;++i)
{
for(int j=0;j<=n;++j)//当前有j个朝上的
{
for(int h=0;h<=k;++h)//抛上去k个中有h个是正的
{
if(n-j>=k)
dp[i+1][j+h]+=dp[i][j]*1.0*c[k][h]*deno[k];
else
dp[i+1][h+n-k]+=dp[i][j]*1.0*c[k][h]*deno[k];
}
}
}
double ans=0.0;
for(int i=0;i<=n;++i)
ans+=dp[m][i]*i;
cout<<fixed<<setprecision(3)<<ans<<endl;
}
return 0;
}