题意:
有n个位置,每个 位置 i 为圈的概率给出,一段序列的值为每段连续圈的长度的平方和,求序列的值的期望
思路:
不是很擅长概率DP,只会用拆分公式的方法。。。
若我们已经知道了长度为 1到(x-1) 的期望F(k) ,我们要求长度为x的期望F(x)
那么此时的期望F(x)写成式子如下
然后就是对这个式子进行整理,先不考虑涉及到 f(k) 的部分 将每个 ( 1 - pi )拆掉,然后合并,得到新式子:
稍作观察,可以发现 g[ x ] 与 g [ x - 1 ] 之间的递推关系:
令大括号里的式子为h[x-1],可以发现h[x]与h[x-1]之间的递推关系:
那么:
然后考虑 f(k) 的部分:
为了方便求递推式,拆分成两个式子:
然后能看出对应的递推式了:
自此,求得答案需要用到的式子的递推公式都已经得到了。。。
代码:
#include<bits/stdc++.h>
using namespace std;
int n;
int main(){
scanf("%d",&n);
double ans = 0 , g = 0 , h = 0 , p , a = 0 , b = 0;
while(n--){
scanf("%lf",&p);
a = a * p + ans;
b = (b + ans)*p;
g = p * ( g + 2 * h + 1 );
ans = g + a - b;
h = ( h + 1 ) * p;
}printf("%.10f\n",ans);
return 0;
}