牛客练习赛26-B 烟花

地址：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;
}