牛客练习赛26B——烟花【动态规划】

链接：https://www.nowcoder.com/acm/contest/180/B
来源：牛客网
 

小a有个烟花，每个烟花代表着互不相同的颜色，对于第个烟花，它有的概率点燃，现在小a要去点燃它们，他想知道产生颜色的期望个数 及 产生恰好产生种颜色的概率

 

输入描述:

第一行两个整数
接下来一行个数，第个数表示第个烟花被点燃的概率

输出描述:

输出有两行
第一行表示产生不同颜色的期望个数

第二行表示产生恰好种颜色的概率

以换行符分割

 

示例1

输入

复制

3 2
0.5 0.25 0.75

输出

复制

1.5000
0.4062

说明

第二问样例解释：

相加得

备注:

对于的数据：
对于的数据：
输出均保留4位小数,若你的答案误差与std不超过即为正确

因为每种颜色都只有一种所以数学期望就是所有颜色的概率之和。dp[i][j]表示i种不同的颜色燃放j种的概率是多少。状态转移方程为dp[i][j-1] += dp[i-1][j] *(1-p[i]),dp[i][j] += dp[i-1][j-1]*p[i];

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 100010;
double dp[MAXN][110],p[MAXN];
int n,k;
int main(){
	while(~scanf("%d %d",&n,&k)){
		double ans = 0.0;
		for(int i = 1; i <= n; i++){
			scanf("%lf",&p[i]);
			ans += p[i];
		}
		dp[0][0] = 1.0;
		for(int i = 0; i <= n-1; i++){
			for(int j = 0; j <= k; j++){
				dp[i+1][j] += dp[i][j] * (1 - p[i+1]);
				dp[i+1][j+1] += dp[i][j] * p[i+1];
			}
		}
		printf("%.4lf\n",ans);
		printf("%.4lf\n",dp[n][k]);
	}
	return 0;
}