洛谷1654 BZOJ4318 OSU! 期望

题目链接

题意：
有一个长度为$n$的序列，每个数字都是$0$或$1$，告诉你每个数是$1$的概率。其中长度为$x$的连续的一段$1$对答案的贡献是$x^3$，求答案的期望。$n<=1e5$，四舍五入保留一位小数。

题解：
我们考虑一个dp。最简单的dp是$n^2$的，但是状态就是$n^2$级别的，所以没什么优化的空间。所以我们考虑直接把期望带进去计算。我们考虑从$i-1$位置到$i$这个位置答案期望增加多少。我们设原来的期望连续$1$的长度是$x$，现在可能变成$x+1$。我们化一下式子可以知道，$(x+1{)}^3-x^3=3x^2+3x+1$，这是权值，再乘上这个位置是$1$的概率，就是期望增加的值。

根据期望的线性性，我们可以把$3x^2+3x+1$拆成三个部分，$1$是常数不用管，那么我们就分别维护出这里$x$的期望值和$x^2$的期望值，就可以算从$i-1$到$i$的答案增加量了。而这两个量也很好维护。$x^2$的期望增加值就用$(x+1{)}^2-x^2$乘上$i$是$1$的概率就好了，$x$的期望增加量直接就是$1\ast i$位置上是$1$的概率。但是注意一下我们更新$i$处的答案用的是$i-1$处的$x$与$x^2$的增加量，因为是之前已有的期望。

这样就做完了，时间复杂度$O(n)$。代码很简单。

代码：

#include <bits/stdc++.h>
using namespace std;

int n;
double x[100010],y[100010],dp[100010],a[100010];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	scanf("%lf",&a[i]);
	for(int i=1;i<=n;++i)
	{
		x[i]=(x[i-1]+1)*a[i];
		y[i]=(y[i-1]+2*x[i-1]+1)*a[i];
		dp[i]=dp[i-1]+(3*y[i-1]+3*x[i-1]+1)*a[i];
	}
	printf("%.1lf\n",dp[n]);
	return 0;
}