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