考虑某一位为1的概率为p,那么它的贡献为:
p*((x+1)^3-x^3)=p*(3*x^2+3*x+1),其中x表示当前连续1的长度,然后开两个变量维护x^2的期望和x的期望然后转移一下就好了。注意平方的期望不等同于期望的平方(就好像平方的平均数不等于平均数的平方)。
AC代码如下:
#include<cstdio>
using namespace std;
int main(){
int i,n; scanf("%d",&n); double ans=0,t1=0,t2=0,x;
for (i=1; i<=n; i++){
scanf("%lf",&x);
ans+=((t1+t2)*3+1)*x; t2=(t2+t1*2+1)*x; t1=(t1+1)*x;
}
printf("%.1f\n",ans);
return 0;
}
by lych
2016.4.18