2017 ICPC乌鲁木齐 A Coins 概率dp

Coins

题意:一开始所有n个硬币都是反面朝上的,每次必须拿k个来抛,抛的人足够聪明,问m次之后向上的硬币的期望。

首先说了这个足够聪明的意思,就是只要向反面的有k个就不会sb地去拿向正面的来抛,想了一会之后就觉得是个概率dp的转移,

然而一开始想漏了个组合数的加权,但在+1的提醒下搞通了,但是分析了下,这是nmk的时间复杂度,

1e6还有个1e3的大T,emmm理论上会TLE的,但结果看网上题解,都是跟自己思路差不多,所以也是很迷。

嗯,转移过程其实很简单,dp[i][j]就是第i次抛了之后j个硬币向上的概率,所以如果n-j>=k很明显,这是全部拿向反面的来抛就行,那么dp[i+1][j+kk]=dp[i][j]*0.5的k次方*k个里面有kk个向正面的组合数

而n-j<的话,那么就得那一些向正面的来抛了,剩下的正面的就j-(k-(n-j))也就是n-k个,那么dp[i+1][n-k+kk]=dp[i][j]*0.5的k次方*k个里面有kk个向正面的组合数

 #include<cstdio>
const int N=;
double cf05[N],c[N][N],dp[N][N];
void init(){
for(int i=;i<=;i++){
c[i][]=1.0;
for(int j=;j<=i;j++){
if(j<=i/) c[i][j]=c[i-][j-]+c[i-][j];
else c[i][j]=c[i][i-j];
}
}
cf05[]=1.0;
for(int i=;i<=;i++) cf05[i]=cf05[i-]*0.5;
}
int main(){
init();
int t,n,m,k;
scanf("%d",&t);
while(t--){
scanf("%d%d%d",&n,&m,&k);
for(int i=;i<=m;i++)
for(int j=;j<=n;j++) dp[i][j]=0.0;
dp[][]=1.0;
for(int i=;i<m;i++){
for(int j=;j<=n;j++){
if(dp[i][j]==0.0) continue;
for(int kk=;kk<=k;kk++){
if(n-j>=k) dp[i+][j+kk]+=dp[i][j]*cf05[k]*c[k][kk];
else dp[i+][n-k+kk]+=dp[i][j]*cf05[k]*c[k][kk];
}
}
}
double ans=0.0;
for(int i=;i<=n;i++) ans+=dp[m][i]*i;
printf("%.3f\n",ans);
}
return ;
}

怪怪的哦

上一篇:1229【MySQL】性能优化之 Index Condition Pushdown


下一篇:JS 鼠标事件大全