版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/C_13579/article/details/82532444
地址:https://www.nowcoder.com/acm/contest/180/B
思路:概率dp,
dp[i][k]:前i种烟花产生k种颜色的概率
dp[i][k]=dp[i-1][k-1]*p+dp[i-1][k]*(1-p);
空间可以滚动数组优化
Code :
#include<iostream>
using namespace std;
const int MAX_M=205;
int n,m;
double dp[5][MAX_M];
int main()
{
while(~scanf("%d%d",&n,&m)){
double ans=0,p=0;
dp[0][0]=1;
int t=0;
for(int i=0;i<n;++i)
{
scanf("%lf",&p);
t=(t+1)%3;
ans+=p;
dp[t][0]=dp[(t+3-1)%3][0]*(1-p);
for(int k=1;k<=m;++k)
dp[t][k]=dp[(t+3-1)%3][k-1]*p+dp[(t+3-1)%3][k]*(1-p);
}
printf("%.4f\n%.4f\n",ans,dp[t][m]);
}
return 0;
}