牛客26

链接：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

转移方程:

dp[i][j] = dp[i-1][j]*(1-p[i])+dp[i-1][j-1]*p[i];
前i件物品中产生j个颜色的概率为前(i-1)个物品产生j个颜色*第i个物品不产生颜色的概率，加上前(i-1)件物品产生(j-1)种颜色*第i个物品产生颜色的概率

ac:

#include<bits/stdc++.h>
using namespace std;
double dp[100001],p[100001],ans1;

int main()
{
    int n,k;scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lf",&p[i]);
        ans1+=p[i];
        dp[0]=1;
    }
    for(int i=1;i<=n;i++)
        for(int j=k;j>=0;j--)
            dp[j]=dp[j]*(1-p[i])+dp[j-1]*p[i];
    printf("%.4lf\n%.4lf",ans1,dp[k]);
}