bzoj 4318 期望的线性性

http://www.lydsy.com/JudgeOnline/problem.php?id=4318

 

就是说，在一段长度为x的1后面加一个1，贡献为(x+1)^3-x^3=3*x^2+3*x+1,只要算出在每个位置结尾的全1子串的长度期望和长度平方的期望即可。而全1子串的长度平方期望也可以考虑每个1的期望贡献2x+1,由长度的期望推出即可。

 

每个位置结尾全1子串长度的平方期望状态转移：x2=x*(x2+2*x1+1);

 

每个位置结尾全1子串长度的期望状态转移：x1=x*(1+x1);

#include<cstdio>
using namespace std;
double x1,x2,x3,x;
int main()
{
    int n;
    scanf("%d",&n);
    x1=x2=x3=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%lf",&x);
        x3+=x*(3*x1+3*x2+1);
        x2=x*(x2+2*x1+1);
        x1=x*(1+x1);
    }
    printf("%.1f\n",x3);
}