BZOJ 4318 OSU!(期望DP )

4318: OSU!

Time Limit: 2 Sec   Memory Limit: 128 MB
Submit: 88   Solved: 64
[ Submit][ Status][ Discuss]

Description

osu 是一款群众喜闻乐见的休闲软件。 

我们可以把osu的规则简化与改编成以下的样子: 

一共有n次操作，每次操作只有成功与失败之分，成功对应1，失败对应0，n次操作对应为1个长度为n的01串。在这个串中连续的 X个1可以贡献X^3 的分数，这x个1不能被其他连续的1所包含（也就是极长的一串1，具体见样例解释） 

现在给出n，以及每个操作的成功率，请你输出期望分数，输出四舍五入后保留1位小数。 

Input

第一行有一个正整数n,表示操作个数。接下去n行每行有一个[0,1]之间的实数，表示每个操作的成功率。 

Output

只有一个实数，表示答案。答案四舍五入后保留1位小数。 

Sample Input

3
0.5
0.5
0.5

Sample Output

6.0

HINT

【样例说明】 

000分数为0，001分数为1，010分数为1，100分数为1，101分数为2，110分数为8，011分数为8，111分数为27，总和为48，期望为48/8=6.0 

N<=100000

考虑每新增一个位置，如果这个位置是 0 ，则贡献为 0 ，否则贡献为 (x+1)3−x3=3x2+3x+1 ，其中 x 为加入之前最长的全1后缀的长度
那么我们只需要维护一个 x 的期望和 x2 的期望即可。

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>
#define ll long long 
using namespace std;
const int MAXN = 100000 + 10;
double p[MAXN], l1[MAXN], l2[MAXN], f[MAXN];
int n;
int main()
{
	while(scanf("%d", &n)!=EOF)
	{
	    for(int i=1;i<=n;i++)
	    {
	        scanf("%lf", &p[i]);
	        l1[i] = (l1[i-1] + 1) * p[i];
	        l2[i] = (l2[i-1] + 2 * l1[i-1] + 1) * p[i];
	        f[i] = f[i-1] + (3 * l2[i-1] + 3 * l1[i-1] + 1) * p[i];
	        
	    }
	    printf("%.1lf\n", f[n]);
	}
	return 0;
}