BZOJ 3450: Tyvj1952 Easy [DP 概率]

传送门

题意：$ox?$组成的序列，$?$等概率为$o\ or\ x$，得分为连续的$o$的长度的平方和，求期望得分

一开始没想出来，原因在于不知道如何记录长度

其实我们同时求得分和长度的期望就好了

$(x+1)^2=x^2+2x+1$

其实就是维护了$x$的期望和$x^2$的期望

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=3e5+5;
inline int read(){
char c=getchar();int x=0,f=1;
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
return x*f;
}
int n;
char s
;
double f
,d
;
int main(){
freopen("in","r",stdin);
n=read();scanf("%s",s+1);
for(int i=1;i<=n;i++){
if(s[i]=='o') d[i]=d[i-1]+1,f[i]=f[i-1]+2*d[i]-1;
else if(s[i]=='x') d[i]=0,f[i]=f[i-1];
else d[i]=(d[i-1]+1)*0.5,f[i]=f[i-1]+(2*d[i-1]+1)*0.5;
}
printf("%.4lf",f
);
}