bzoj 4318 期望的线性性

最新推荐文章于 2019-08-05 19:55:10 发布

原创最新推荐文章于 2019-08-05 19:55:10 发布 · 247 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

本文介绍了一种计算字符串中全1子串长度及其平方的期望值的方法，并提供了具体的算法实现。通过状态转移方程，可以高效地解决在每个位置结尾的全1子串的长度及长度平方的期望值问题。

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);
}