Po姐的思路:
每新增一个位置,如果这个位置是0 ,则贡献为0 ,
否则贡献为
(x+1)3−x3=3x2+3x+1
(
x
+
1
)
3
−
x
3
=
3
x
2
+
3
x
+
1
,x为加入之前最长的全1后缀的长度
现在这个问题变成了期望问题,那么我们只需要维护一个
x
x
的期望和 的期望即可。
注意 (x+1)2 ( x + 1 ) 2 也要转换为 x2+2x+1 x 2 + 2 x + 1 解决
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100010;
double p[maxn],p1[maxn],p2[maxn],f[maxn];
//贡献:(x+1)^3-x^3=3x^2+3x+1
int main()
{
int n;scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%lf",&p[i]);
p1[i]=(p1[i-1]+1)*p[i];//x+1
p2[i]=(p2[i-1]+2*p1[i-1]+1)*p[i];//(x+1)^2=x^2+2x+1
f[i]=f[i-1]+(3*p2[i-1]+3*p1[i-1]+1)*p[i];
}
printf("%.1lf\n",f[n]);
return 0;
}