hihocode 1506 投掷硬币

描述

小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 
#include 
#include 
#include  
#include 
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;
}

你可能感兴趣的:(dp)