题目
小a有
n
n
个烟花,每个烟花代表着互不相同的颜色,对于第个烟花,它有
pi
p
i
的概率点燃,现在小a要去点燃它们,他想知道产生颜色的期望个数 及 产生恰好产生
k
k
种颜色的概率
输入描述:
第一行两个整数,接下来一行
n
n
个数,第个数表示第
i
i
个烟花被点燃的概率
输出描述:
输出有两行
第一行表示产生不同颜色的期望个数,第二行表示产生恰好种颜色的概率,以换行符分割
示例1
输入
3 2
0.5 0.25 0.75
输出
1.5000
0.4062
分析
第一问就是double版的a+b,很好,来说第二问,我们设表示到第i个数有j个颜色的概率,转移的话可以考略是否会燃,进行两方面考虑即可。
上代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
double num[100010],ans1=0.0,ans2[100010][210];
int n,k;
int main(){
scanf("%d%d",&n,&k);
ans2[0][0]=1.0;
for(int i=1;i<=n;i++){
scanf("%lf",&num[i]);
ans1+=num[i];
for(int j=0;j<=k;j++){
if(j)
ans2[i][j]+=ans2[i-1][j-1]*num[i];
ans2[i][j]+=ans2[i-1][j]*(1-num[i]);
}
}
printf("%.4lf\n%.4lf",ans1,ans2[n][k]);
return 0;
}