牛客练习赛26。B

题目

小a有$n$个烟花，每个烟花代表着互不相同的颜色，对于第$i$个烟花，它有$p_i$的概率点燃，现在小a要去点燃它们，他想知道产生颜色的期望个数 及 产生恰好产生$k$种颜色的概率
输入描述:
第一行两个整数$n,k$，接下来一行$n$个数，第$i$个数表示第$i$个烟花被点燃的概率
输出描述:
输出有两行
第一行表示产生不同颜色的期望个数，第二行表示产生恰好种颜色的概率，以换行符分割

示例1
输入
3 2
0.5 0.25 0.75
输出
1.5000
0.4062

分析

第一问就是double版的a+b，很好，来说第二问，我们设$f[i,j]$表示到第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;
}