bzoj3450 (概率dp)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_37632935/article/details/81772138

题意:有一个长度为 n 的字符串，由 o,x,? 三种字符组成。? 代表 o,x 各有 50% 概率。求连续 o长度的平方和的期望。

思路:由平方公式可得,那么就可以设i之前连续的o的长度期望为f[i].

如果s[i]='o'那么答案贡献为f[i-1]*2+1,f[i]=f[i-1]+1.

如果s[i]='x'f[i]=0,贡献也为0

如果s[i]='？'贡献为(f[i-1]*2+1)/2,f[i]=(f[i-1]+1)/2.

#include <bits/stdc++.h>
using namespace std;
double f[300005];
int n;double ans;char s[300005];
int main()
{
	scanf("%d",&n);
	scanf("%s",s+1);
	for(int i=1;i<=n;i++)
	{
		if(s[i]=='o') ans+=2*f[i-1]+1,f[i]=f[i-1]+1;
		else if(s[i]=='x')f[i]=0;
		else ans+=f[i-1]+1.0/2.0,f[i]=f[i-1]*0.5+0.5;
	}
	printf("%.4lf\n",ans);
	return 0;
}