Codeforces Round #146 (Div. 2) D. Let's Play Osu! comb数平方和的数学期望

题目:
D. Let's Play Osu!
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

You're playing a game called Osu! Here's a simplified version of it. There aren clicks in a game. For each click there are two outcomes: correct or bad. Let us denote correct as "O", bad as "X", then the whole play can be encoded as a sequence ofn characters "O" and "X".

Using the play sequence you can calculate the score for the play as follows: for every maximal consecutive "O"s block, add the square of its length (the number of characters "O") to the score. For example, if your play can be encoded as "OOXOOOXXOO", then there's three maximal consecutive "O"s block "OO", "OOO", "OO", so your score will be 22 + 32 + 22 = 17. If there are no correct clicks in a play then the score for the play equals to 0.

You know that the probability to click the i-th(1 ≤ i ≤ n) click correctly ispi. In other words, thei-th character in the play sequence haspi probability to be "O",1 - pi to be "X". You task is to calculate the expected score for your play.

Input

The first line contains an integer n (1 ≤ n ≤ 105) — the number of clicks. The second line containsn space-separated real numbers p1, p2, ..., pn(0 ≤ pi ≤ 1).

There will be at most six digits after the decimal point in the given pi.

Output

Print a single real number — the expected score for your play. Your answer will be considered correct if its absolute or relative error does not exceed10 - 6.

Sample test(s)
Input
3
0.5 0.5 0.5
Output
2.750000000000000
Input
4
0.7 0.2 0.1 0.9
Output
2.489200000000000
Input
5
1 1 1 1 1
Output
25.000000000000000
Note

For the first example. There are 8 possible outcomes. Each has a probability of 0.125.

  • "OOO"  → 32 = 9;
  • "OOX"  → 22 = 4;
  • "OXO"  → 12 + 12 = 2;
  • "OXX"  → 12 = 1;
  • "XOO"  → 22 = 4;
  • "XOX"  → 12 = 1;
  • "XXO"  → 12 = 1;
  • "XXX"  → 0.

So the expected score is

 

题意,给出每个位置O出现的概率,求连续O的个数的平方和的数学期望。

思路:(不是我想出来的,大牛教的)设ai为第i段连续O的长度,∑ai^2 = ∑[ ai+ ai*(ai-1) ] = ∑ ai*(ai-1) + ∑ai = ∑ C(ai, 2)*2 + ∑ai

这样就转化为长度大于1的连续段数*2+O的个数

例如:OOXOXOOO,它的值是4+1+9=14

转化后计算则为4*2+6=14 (4段长度大于1的连续段分别为1~2、5~6、5~7、6~7,再加上O的个数6)

dp[i]表示以i结尾的连续O的段数的期望(这里包括长度为1的,上例中,dp[1]=1,dp[2]=2,dp[7]=2,dp[8]=3),状态转移则为dp[i]=dp[i-1]*p[i]+p[i];

 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=100005;
double p[maxn], dp[maxn];
int n;
int main(){
    //freopen("1.txt", "r", stdin);
    int i;
    while(scanf("%d", &n)!=EOF){
        double ans=0;
        for(i=1; i<=n; i++){
            scanf("%lf", &p[i]);
            ans+=p[i];
        }
        dp[0]=0;
        for(i=1; i<=n; i++){
            dp[i]=dp[i-1]*p[i]+p[i];
            ans+=(dp[i]-p[i])*2;
        }
        printf("%.7f\n", ans);
    }
    return 0;
}


 

 

 

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值