bzoj 4318 OSU! 期望dp

---

4318: OSU!

Description

osu 是一款群众喜闻乐见的休闲软件。
我们可以把osu的规则简化与改编成以下的样子:
一共有n次操作，每次操作只有成功与失败之分，成功对应1，失败对应0，n次操作对应为1个长度为n的01串。在这个串中连续的 X个1可以贡献X^3 的分数，这x个1不能被其他连续的1所包含（也就是极长的一串1，具体见样例解释）
现在给出n，以及每个操作的成功率，请你输出期望分数，输出四舍五入后保留1位小数。

Input

第一行有一个正整数n,表示操作个数。接下去n行每行有一个[0,1]之间的实数，表示每个操作的成功率。

Output

只有一个实数，表示答案。答案四舍五入后保留1位小数。

Sample Input

3

0.5

0.5

0.5

Sample Output

6.0

HINT

【样例说明】

000分数为0，001分数为1，010分数为1，100分数为1，101分数为2，110分数为8，011分数为8，111分数为27，总和为48，期望为48/8=6.0

N<=100000

---

令$E[i]$表示以位置i结尾的$'1'$串相比以位置i-1结尾的$'1'$串新增分的期望。(i这个位置为1时的贡献值)
$ans=\sum_{i=1}^nE[i]*p[i]$
$∵(x+1)^3-x^3=3x^2+3x+1$(x为原来的长度，x+1为新长度)
$∴E[i]=E(3x^2+3x+1)=3E(x^2)+3E(x)+1$
故需要维护$E(x^2)$和$E(x)$

同理可得$E_x[i]=p[i]*(E_x[i-1]+1)$
$E_{x2}[i]=p[i]*(E_{x2}[i-1]+2E_x[i-1]+1);$

---

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<iomanip>
using namespace std;

#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
#define mes(a,x,s)  memset(a,x,(s)*sizeof a[0])
#define mem(a,x)  memset(a,x,sizeof a)
#define ysk(x)  (1<<(x))
typedef long long ll;
const int INF =0x3f3f3f3f;
const int maxn= 100000   ;
int n;
double Ex[maxn+10],Ex2[maxn+10],p[maxn+10];
int main()
{
   std::ios::sync_with_stdio(false);
   while(cin>>n)
   {
       double ans=0;
       Ex[0]=Ex2[0]=0;
       for1(i,n)
       {
           cin>>p[i];
           ans+=p[i]*(1+ 3*Ex[i-1]+ 3*Ex2[i-1]);
           Ex[i]=(Ex[i-1]+1)*p[i];
           Ex2[i]=p[i]*(Ex2[i-1]+1+2*Ex[i-1]);
       }
       cout<<fixed<<setprecision(1)<<ans<<endl;
   }

   return 0;
}