#include <algorithm>
#include <iostream>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define xin(a) scanf("%d", &a)
using namespace std;
const int maxn = 100000 + 5;
double dp[maxn][202];//到i个时已点燃j个;
int main()
{
double sum=0;
int n, k, i, j;
double a[maxn];
cin>>n>>k;
for (i=1;i<=n;i++)
{
cin>>a[i];
sum+=a[i];
}
dp[0][0]=1;
dp[0][1]=0;
for(i=1;i<=n;i++)
{
dp[i][0]=dp[i-1][0]*(1-a[i]);//!!!!!!
for(int j=1;j<=k; j++)
dp[i][j]=dp[i-1][j-1]*a[i]+dp[i-1][j]*(1-a[i]);
}
printf("%.4f\n%.4f\n",sum,dp[n][k]);
return 0;
}
解题思路
第一次看到概率dp吼吼
第一问直接相加,第二问用dp解决,dp[i][j]表示到第i个时已经点燃j个的概率,
dp[i][j]=dp[i-1][j-1]×a[i]+dp[i-1][j]×(1-a[i]);
dp[i-1][j-1]×a[i]表示当前烟花点燃的概率,j+1.
dp[i-1][j]×(1-a[i]);表示当前烟花未点燃
特别注意dp[i][0]需要特别处理
(滚动数组什么的..下次一定?)

京公网安备 11010502036488号