烟花--牛客练习赛26

此题链接

上代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int main()
{
    int n,k,o=0;
    cin>>n>>k;
    double p[100001],f[2][100001],sum=0;
    for(int i=0;i<n;i++)
    {
        cin>>p[i];
        sum+=p[i];
    }
    f[o][1]=p[0];
    f[o][0]=1-p[0];
    for(int i=2;i<=n;i++)
    {
        o=1-o;
        for(int j=0;j<=min(i,k);j++)
        {
            f[o][j]=f[1-o][j-1]*p[i-1]+f[1-o][j]*(1-p[i-1]);
        }
    }
    printf("%.4lf\n%.4lf",sum,f[o][k]);
}

第一问高中知识便可搞定；

第二问要用到滚动数组和动态规划；

针对第二问设f[i][j]为前i个恰好有j个烟花被点燃的概率

则有f[i][j]=f[i-1][j-1]*p[i]+f[i-1][j]*(1-p[i]);

另外滚动数组的变量常用o(圆圆的(｀・ω・´))

一般o^=1或者o=1-o

最后你细心一点可能会发现数组越界了！！

其实C++允许数组越界而且Key=-1时值为零；

故不影响结果；