题目地址:https://hihocoder.com/problemset/problem/1506
时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
小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问题 ,注意边界
dp[i][j] = dp[i-1][j] *np[i] + dp[i-1][j-1] *p[i];
c++代码如下:
#include<iostream>
using namespace std;
float dp[1024][1024]={0};
int main(){
int M,N;
float p[1024]={0};
float np[1024]={0};
cin>>N>>M;
for(int i=1;i<=N;i++){
cin>>p[i];
np[i] = 1-p[i];
}
arr[1][0] = np[1];
arr[1][1] = p[1];
for(int i = 2;i<=N;i++){
for(int j=0;j<=M;j++){
dp[i][j] = dp[i-1][j] *np[i] ;
if(j>0){
dp[i][j]+=dp[i-1][j-1] *p[i];
}
}
}
cout<<dp[N][M]<<endl;
}