链接: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[1][0] = (1-p[1]);dp[1][1] = p[1];
递推式就是
dp[i][j]=dp[i−1][j]∗(1−p[i])+dp[i][j−1]∗p[i]
d
p
[
i
]
[
j
]
=
d
p
[
i
−
1
]
[
j
]
∗
(
1
−
p
[
i
]
)
+
d
p
[
i
]
[
j
−
1
]
∗
p
[
i
]
复杂度是O(nk)
代码如下:
#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 1e5+10;
const int MAX_K = 2e2+10;
double dp[MAX_N][MAX_K],p[MAX_N];
int main(void){
int N,K;
scanf("%d%d",&N,&K);
double sum = 0;
for(int i=1;i<=N;++i){
scanf("%lf",&p[i]);
sum += p[i];
}
dp[1][0] = 1-p[1];dp[1][1] = p[1];
for(int i=2;i<=N;++i){
dp[i][0] = dp[i-1][0]*(1-p[i]);
for(int j=1;j<=min(i,K);++j){
dp[i][j] = dp[i-1][j]*(1-p[i])+dp[i-1][j-1]*p[i];
}
}
printf("%.4lf\n",sum);
printf("%.4lf\n",dp[N][K]);
return 0;
}