【BZOJ 4318】[概率DP]OSU!

最新推荐文章于 2019-08-29 15:59:33 发布

ORZSunIsMe

最新推荐文章于 2019-08-29 15:59:33 发布

阅读量440

点赞数 1

分类专栏：动态规划概率dp 文章标签：概率dp bzoj

本文链接：https://blog.csdn.net/qq_35479641/article/details/61411316

版权

动态规划同时被 2 个专栏收录

28 篇文章 0 订阅

订阅专栏

概率dp

3 篇文章 0 订阅

订阅专栏

题目描述

给出 $n$ 和 $p_i|i\leq n$ ， $p_i$ 表示在第 $i$ 位上出现1的概率，对于每一段连续的1，它的价值是 $len^3$ ，求该数列的期望价值。

题目分析

首先对于当前位置，若为0，则贡献为0；若为1，则贡献为 $len_i^3-len_{i-1}^3=3\times len_{i-1}^2+3\times len_{i-1}+1$ 。
根据概率的计算规则，我们可以知道len的期望 $len_i=(len_{i-1}+1)\times p_i+(0\times (1-p_i))=(len_{i-1}+1)\times p_i$
$len_i^2=(len_{i-1}^2+2\times len_{i-1}+1)\times p_i$
那么若设 $dp_i$ 为前 $i$ 个数的期望价值，就有 $dp_i=dp_{i-1}+(3\times len_{i-1}^2+3\times len_{i-1}+1)\times p_i$

代码

/**************************************************************
    Problem: 4318
    User: szpszp
    Language: C++
    Result: Accepted
    Time:160 ms
    Memory:3632 kb
****************************************************************/

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;

#define MAXN 100000
#define MAXM
#define INF 0x3f3f3f3f
typedef long long int LL;

template<class T>
void Read(T &x){
    x=0;char c=getchar();bool flag=0;
    while(c<'0'||'9'<c){if(c=='-')flag=1;c=getchar();}
    while('0'<=c&&c<='9'){x=x*10+c-'0';c=getchar();}
    if(flag)x=-x;
}

double l[MAXN+10];
double l2[MAXN+10];
double dp[MAXN+10];
int n;

int main(){
    Read(n);
    double x;

    for(int i=1;i<=n;++i){
        scanf("%lf",&x);
        l[i]=(l[i-1]+1)*x;
        l2[i]=(l2[i-1]+2*l[i-1]+1)*x;
        dp[i]=dp[i-1]+(3*l2[i-1]+3*l[i-1]+1)*x;
    }

    printf("%0.1lf\n",dp[n]);
}