投掷硬币

题目名称：投掷硬币

题目链接：

描述

小Hi有一枚神奇的硬币。已知第i次投掷这枚硬币时，正面向上的概率是Pi。

现在小Hi想知道如果总共投掷N次，其中恰好M次正面向上的概率是多少。

输入

第一行包含两个整数N和M。

第二行包含N个实数$P_1,P_2,...P_N$。

对于30%的数据，1 <= N <= 20

对于100%的数据，1 <= N <= 1000, 0 <= M <= N, 0 <= $P_i$ <= 1

输出

输出一行一个实数表示恰好M次正面向上的概率。注意行末需要包含一个换行符’\n’。

输出与标准答案误差在0.001以内都被视为正确。

样例输入

2 1
0.5 0.5

样例输出

0.500000

解题思路

概率动态规划问题，使用使用f[i][[j]表示投掷i枚硬币，其中j面向上的概率，所以有如下思路，i枚硬币中j面向上的概率为第 i-1枚硬币中j-1面向上的概率 * 第i枚硬币向上的概率+第i-1枚硬币中j 面向上给概率 * 第i枚硬币向下的概率 ，即
$$f[i][j]=f[i-1][j]\ast (1-Pi)+f[i-1][j-1]\ast Pi$$

完整代码

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
#define N 1010
double f[N][N];
int n,m;
double p[N];

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>p[i];
	}
	f[1][1]=p[1];
	f[1][0]=1-p[1];
	for(int i=2;i<=n;i++){
		f[i][0]=f[i-1][0]*(1-p[i]);
		for(int j=1;j<=m;j++){
			f[i][j]=f[i-1][j]*(1-p[i])+f[i-1][j-1]*p[i];
		}
	}
	printf("%.6lf\n",f[n][m]);
	return 0;
}