【P1365】【BZOJ4318】WJMZBMR打osu! / Easy 期望Dp

题目描述

某一天WJMZBMR在打osu~~~但是他太弱逼了，有些地方完全靠运气:(
我们来简化一下这个游戏的规则
有n次点击要做，成功了就是o，失败了就是x，分数是按comb计算的，连续a个comb就有a*a分，comb就是极大的连续o。
比如ooxxxxooooxxx，分数就是2*2+4*4=4+16=20。
Sevenkplus闲的慌就看他打了一盘，有些地方跟运气无关要么是o要么是x，有些地方o或者x各有50%的可能性，用?号来表示。
比如oo?xx就是一个可能的输入。 那么WJMZBMR这场osu的期望得分是多少呢？
比如oo?xx的话，?是o的话就是oooxx => 9，是x的话就是ooxxx => 4
期望自然就是(4+9)/2 =6.5了

输入输出格式

输入格式：
第一行一个整数n，表示点击的个数
接下来一个字符串，每个字符都是ox？中的一个
输出格式：
一行一个浮点数表示答案
四舍五入到小数点后4位
如果害怕精度跪建议用long double或者extended

输入输出样例

输入样例#1：
4
????
输出样例#1：
4.1250

题解

我们设$g[i]$表示当前到第$i$个字符的期望长度，$f[i]$为到第i个字符的期望得分
转移方程非常容易，分三种情况：
1.$s[i]='x',f[i]=f[i-1],g[i]=0$如果当前是$'x'$，那么长度就被清空，得分也和上一个一样
2.$s[i]='o',f[i]=f[i-1]+2*g[i-1]=1,g[i]=g[i-1]+1$，如果当前是$'o'$，得分展开完全平方公式就可以$(g[i-1]+1)^2=g[i-1]^2+2*g[i-1]+1=f[i-1]+2*g[i-1]+1$
3.$s[i]='?',g[i]=(g[i-1]+1)/2,f[i]=f[i-1]+(2*g[i-1]+1)/2$，如果当前是$'?'$，期望长度为$g[i-1]+1$,概率为$\frac{1}{2}$,$期望长度就是(g[i-1]+1)*\frac{1}{2}$。现在考虑得分当前的得分增加的概率为$\frac{1}{2}$，所以$f[i]=f[i-1]+(2*g[i-1]+1)/2$

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
const int MAXN=1e7;
using namespace std;
int blablabla;
char s[MAXN];
double g[MAXN],f[MAXN];
int main()
{
    scanf("%d",&blablabla);
    scanf("%s",s);
    int len=strlen(s);
    for(int i=1;i<=len;i++)
    {
        if(s[i-1]=='x')
        {
            f[i]=f[i-1],g[i]=0;
        }
        else if(s[i-1]=='o')
        {
            f[i]=f[i-1]+2*g[i-1]+1,g[i]=g[i-1]+1;
        }
        else if(s[i-1]=='?')
        {
            f[i]=f[i-1]+((2*g[i-1]+1))/2,g[i]=(g[i-1]+1)/2;
        }
    }
    printf("%.4f",f[len]);
    return 0;
}