bzoj 4318 OSU! 题解（期望）

原题链接：
洛谷
B站OJ

题意简述

给定$n$，有一个长度为$n$的$01$串，其中第$i$个位置$=1$的概率为$p_i$。定义一个$01$序列的得分为$l^3$，其中$l$是最长连续$1$的长度。求得分的期望值。

数据

输入

第一行一个正整数$n(n<=1e5)$。接下来$n$行每行一个小数，表示每个位置是$1$的概率。

输出

输出一行一个整数，表示得分的期望值。

样例

输入
3
0.5
0.5
0.5
输出
6.0

思路

众所周知期望的线性性：$E(x+y)=E(x)+E(y)$。
同理也容易推得：$E(x+y+z)=E(x)+E(y)+E(z)$
$E(a_1+a_2...+a_k)=E(a_1)+E(a_2)...+E(a_k)$
这个珂以用来干嘛呢？比如我们要求$x$的期望，我们知道$x$有几种情况，然后我们知道每种情况的期望，那么我们只要把这些期望加起来，就是$x$的期望了。

这种题，直接枚举每个位置的$01$情况肯定是不珂取的，所以。。。递推？

尝试一下吧。我们设$f(i)$为：考虑前$i$个位置的答案。那么我们要用$f(i-1)$求出$f(i)$
为了方便思考（即配合期望的概念），我们设前$i-1$个位置有$k$种取值，其中每种取值的概率是$q_1,q_2...q_k$，最长连续$1$的长度是$v_1,v_2...v_k$，那么前$i-1$个位置的答案（即三次方的期望）就是
$f(i-1)=\sum _{j=1}^k{q}_j\ast v_j^3$。

我们现在有$p_i$的概率会使第$i$个位置是$1$。此时的每个$v_i$都要$+1$变成：$p_i\sum _{j=1}^k{q}_j(v_j+1{)}^3$，用$f(i-1)$表示，也就是
$p_i\sum _{j=1}^k{q}_j(v_j^3+3v_j^2+3v_j+1)$
$=p_i(f(i-1)+\sum _{j=1}^k{q}_{j}3v_j^2+\sum _{j=1}^k{q}_{j}3v_j+\sum _{j=1}^k{q}_j)$
我们知道，一个变量所有取值的概率和肯定是$1$，即$\sum _{j=1}^k{q}_j=1$
原式
$=p_i(f(i-1)+\sum _{j=1}^k{q}_{j}3v_j^2+\sum _{j=1}^k{q}_{j}3v_j+1)$
另外还有$1-p_i$的概率会使第$i$个位置是$0$。此时$v_i$不变。此时的答案：
$(1-p_i)\sum _{j=1}^k{q}_j{v}_j^3$
$=(1-p_i)f(i-1)$

那么，根据线性性，我们把这些情况加起来，就是我们要求的期望值$f(i)$，即
$p_i(f(i-1)+\sum _{j=1}^k{q}_{j}3v_j^2+\sum _{j=1}^k{q}_{j}3v_j+1)+(1-p_i)f(i-1)$
（其中红色的$+$是为了区分两个情况）

我们会发现，右边很好求，但是左边有两个很烦人的东西：
平方的期望（即$\sum _{j=1}^k{q}_i{v}_i^2$） 和 一次方的期望（即$\sum _{j=1}^k{q}_i{v}_i$）
但这两个东西也不是不能求，和三次方类似的推一下转移即珂。我们上面已经推出来三次方的转移了，这两个还不好推？

如果您真的感觉不好推，那么我深感抱歉，并在这里给出推导过程。

我们设前$i$个位置上一次方的期望为$l_1(i)$，二次方的期望为$l_2(i)$。

显然,

$l_1(i)=p_i\sum _{j=1}^k{q}_j(v_j+1)=p_i(f(i-1)+1)$

$l_2(i)=p_i\sum _{j=1}^k{q}_j(v_j+1{)}^2=p_i\sum _{j=1}^k{q}_j(v_j^2+2v_j+1)$

$=p_i(l_2(i-1)+2l_1(i-1)+1)$

然后就珂以求得

$f(i)=p_i(f(i-1)+3l_2(i-1)+3l_1(i-1)+1)+(1-p_i)f(i-1)$

代码（式子长，代码短）：

#include<bits/stdc++.h>
using namespace std;
namespace Flandle_Scarlet
{
    #define real double
    #define N 100100
    int n;
    real p[N];
    real l1[N],l2[N],f[N];
    void Input()
    {
        scanf("%d",&n);
        for(int i=1;i<=n;++i)
        {
            scanf("%lf",&p[i]);
        }
    }
 
    void Soviet()
    {
        for(int i=1;i<=n;++i)
        {
            l1[i]=p[i]*(l1[i-1]+1);
            l2[i]=p[i]*(l2[i-1]+2*l1[i-1]+1);
            f[i]=p[i]*(f[i-1]+3*l2[i-1]+3*l1[i-1]+1)+(1-p[i])*f[i-1];
            //上面的式子
        }
        printf("%.1f\n",f[n]);
    }
    void IsMyWife()
    {
        if (0)
        {
            freopen("","r",stdin);
            freopen("","w",stdout);
        }
        Input();
        Soviet();
    }
};
int main()
{
    Flandle_Scarlet::IsMyWife();
    return 0;
}

回到总题解界面