（期望dp）洛谷P1654OSU!

洛谷P1654OSU!

思路：

利用期望的性质：
$E(X+Y)=E(X)+E(Y)$
$E(C)=C，C为常数$
设连续$i$个数为$1$，贡献的期望为$E(i^3)$。可得：
$E(i)=(E(i-1)+1)\times p_i$
$E(i^2)=(E((i-1{)}^2)+2E(i-1)+1)\times p_i$
$E(i^3)=(E((i-1{)}^3)+3E((i-1{)}^2)+3E(i-1)+1)\times p_i$
设$d{p}_{k,i}$为前$i$个数的贡献期望,$k$表示连续个数为$X$时贡献为$X^k$，可得：
$d{p}_{1,i}=E(i)=(d{p}_{1,i-1}+1)\times p_i$
$d{p}_{2,i}=E(i^2)=(d{p}_{2,i-1}+2\times d{p}_{1,i-1}+1)\times p_i$
$\begin{array}{rl}d{p}_{3,i}& =E(i^3)+d{p}_{3,i-1}\times (1-p_i)\\ & =(E((i-1{)}^3)+3E((i-1{)}^2)+3E(i-1)+1)\times p_i+d{p}_{3,i-1}\times (1-p_i)\\ & =d{p}_{3,i-1}+(3\times d{p}_{2,i-1}+3\times d{p}_{1,i-1}+1)\times p_i\end{array}$
输出$d{p}_{3,n}$即可。

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z)	for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
double dp[5][N],p[N];
int main()
{
	int n,i;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%lf",&p[i]);
	for(i=1;i<=n;i++)
	{
		dp[1][i]=(dp[1][i-1]+1)*p[i];
		dp[2][i]=(dp[2][i-1]+2*dp[1][i-1]+1)*p[i];
		dp[3][i]=dp[3][i-1]+(3*dp[2][i-1]+3*dp[1][i-1]+1)*p[i];
	}
	printf("%.1lf\n",dp[3][n]);
	return 0;
}