（期望dp）洛谷P1654OSU!

最新推荐文章于 2022-08-30 10:56:21 发布

会飞的小蛇

最新推荐文章于 2022-08-30 10:56:21 发布

阅读量232

点赞数

分类专栏： # dp

本文链接：https://blog.csdn.net/Z7784562/article/details/110151739

版权

dp 专栏收录该内容

44 篇文章 0 订阅

订阅专栏

洛谷P1654OSU!

思路：

利用期望的性质：
$E (X + Y) = E (X) + E (Y)$
$E (C) = C ， C 为常数$
设连续 $i$ 个数为 $1$ ，贡献的期望为 $E(i^3)$ 。可得：
$E(i)=(E(i-1)+1)\times p_i$
$E(i^2)=(E((i-1)^2)+2E(i-1)+1)\times p_i$
$E(i^3)=(E((i-1)^3)+3E((i-1)^2)+3E(i-1)+1)\times p_i$
设 $dp_{k,i}$ 为前 $i$ 个数的贡献期望, $k$ 表示连续个数为 $X$ 时贡献为 $X^k$ ，可得：
$dp_{1,i}=E(i)=(dp_{1,i-1}+1)\times p_i$
$dp_{2,i}=E(i^2)=(dp_{2,i-1}+2\times dp_{1,i-1}+1)\times p_i$
$\begin{aligned}dp_{3,i}&=E(i^3)+dp_{3,i-1}\times(1-p_i)\\&=(E((i-1)^3)+3E((i-1)^2)+3E(i-1)+1)\times p_i+dp_{3,i-1}\times(1-p_i)\\&=dp_{3,i-1}+(3\times dp_{2,i-1}+3\times dp_{1,i-1}+1)\times p_i\end{aligned}$
输出 $dp_{3,n}$ 即可。

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z)	for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
double dp[5][N],p[N];
int main()
{
	int n,i;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%lf",&p[i]);
	for(i=1;i<=n;i++)
	{
		dp[1][i]=(dp[1][i-1]+1)*p[i];
		dp[2][i]=(dp[2][i-1]+2*dp[1][i-1]+1)*p[i];
		dp[3][i]=dp[3][i-1]+(3*dp[2][i-1]+3*dp[1][i-1]+1)*p[i];
	}
	printf("%.1lf\n",dp[3][n]);
	return 0;
}