hihocode 1506 投掷硬币

来源：互联网发布：金蝶软件技术服务编辑：程序博客网时间：2024/05/01 22:55

描述

小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 10.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;}

0 0