Description
Input
仅有一行,该行给出依次两个正整数N, M,分别表示宝石的个数和宝石在变化时可能变成的颜色种类数。
Output
应仅有一行,该行给出一个实数E(R),表示圆环的“美观程度”的期望值。
Sample Input
8 1
Sample Output
8.00000
题解:
先考虑序列.
设f[i][0/1]表示前i个珠子,最后1个珠子和第1个珠子颜色不同(相同)的期望值.
设g[i]表示i个珠子连续1个颜色的概率.
显然
f[i][0]=(i-j)*g[i-j]*(f[j][0]*(m-2)/m+f[j][1]*(m-1)/m);
f[i][1]=(i-j)*g[i-j]*f[j][0]*1/m;
初始f[0][1]=1;
然后处理环形.
首先将g[n]*n累加进答案.
我们可以枚举第一段有多长.如果长度为x,那么就可以有x个位置.
所以对于每个x,将x*x*f[n-x][0]*g[x]累加进答案.
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define N 210
using namespace std;
int n,m;
double ans,f[N][2],g[N];
int main(){
scanf("%d%d",&n,&m);g[1]=1;f[0][1]=1;
for(int i=2;i<=n;i++)g[i]=g[i-1]*1.0/(double)m;
for(int i=0;i<=n;i++)
for(int j=i+1;j<=n;j++){
f[j][0]+=(j-i)*g[j-i]*(f[i][0]*(double)(m-2.0)/m+f[i][1]*(double)(m-1.0)/m);
f[j][1]+=(j-i)*g[j-i]*f[i][0]*1.0/(double)m;
}
ans=g[n]*n;
for(int i=1;i<n;i++) ans+=(double)i*(double)i*f[n-i][0]*g[i];
return printf("%.5lf",ans),0;
}