描述
小Hi有一枚神奇的硬币。已知第i次投掷这枚硬币时,正面向上的概率是Pi。
现在小Hi想知道如果总共投掷N次,其中恰好M次正面向上的概率是多少。
输入
第一行包含两个整数N和M。
第二行包含N个实数P1, P2, … PN。
对于30%的数据,1 <= N <= 20
对于100%的数据,1 <= N <= 1000, 0 <= M <= N, 0 <= Pi <= 1
输出
输出一行一个实数表示恰好M次正面向上的概率。注意行末需要包含一个换行符’\n’。
输出与标准答案误差在0.001以内都被视为正确。
样例输入
2 1
0.5 0.5
样例输出
0.500000
dp[i][j]表示 i次投,j次正面朝上。
#include <iostream>
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e3+50;
double a[maxn], dp[maxn][maxn];
int main(){
int n,m;
while( scanf("%d%d",&n,&m) != EOF){
for(int i=1;i<=n;i++) scanf("%lf",a+i);
dp[0][0] = 1, dp[0][1] = 0;
for(int i=1;i<=n;i++){
dp[i][0] = dp[i-1][0]*(1.0-a[i]);
for(int j=1;j<=i;j++){
dp[i][j] = dp[i-1][j]*(1.0-a[i]) + dp[i-1][j-1]*a[i];
}
}
printf("%lf\n",dp[n][m]);
}
return 0;
}