bzoj3450 Tyvj1952 Easy 简单的数学期望

题目大意：一个长度为n（n<300，000）的序列，每个地方可以是o,x,?中的一个。‘？’表示x和o的概率各是50%。一个序列的权值是所有连续的‘o’长度的平方和，问给出序列的期望权值是多少。
例如
oo?xxo
如果’？’是’o’，则权值为3²+1²=10；
如果’?’是‘x’,则权值为2²+1²=5；
期望就是（10+5）/2=7.5；

=====================================================
w[i]为前i位的期望权值。
d[i]为从第i位向前扩展的期望长度。
如果是o w[i]=w[i-1]+2*d[i-1]+1; d[i]=d[i-1]+1;
如果是x w[i]=w[i-1]; d[i]=0;
如果是？ w[i]=(w[i-1]+dp[i-1]*2+1+w[i-1])/2; d[i]=(d[i-1]+1)/2;
最后输出w[n]

#include<cstdio>
#include<iostream>
using namespace std;
double dp[300005];
double w[300005];
char a[300005];
int main()
{
    int n;
    scanf("%d",&n);
    scanf("%s",a+1);
    double ans=0;
    for(int i=1;i<=n;i++)
    {
        if(a[i]=='o')
        {
            dp[i]=dp[i-1]+1;
            w[i]=w[i-1]+dp[i-1]*2+1;
        }
        if(a[i]=='x')
        {
            dp[i]=0;
            w[i]=w[i-1];
        }
        if(a[i]=='?')
        {
            w[i]=(w[i-1]+dp[i-1]*2+1+w[i-1])/2;
            dp[i]=(dp[i-1]+1)/2;
        }
    }
    printf("%.4lf",w[n]);
}