BZOJ2201-彩色圆环

2201: 彩色圆环
Time Limit: 10 Sec Memory Limit: 259 MB
$Description$

$Input$
仅有一行，该行给出依次两个正整数N, M，分别表示宝石的个数和宝石在变化时可能变成的颜色种类数。
$Output$
应仅有一行，该行给出一个实数E(R)，表示圆环的“美观程度”的期望值。
$Sample$ $Input$
8 1
$Sample$ $Output$
8.00000
【数据规模和约定】
100%的数据满足1 ≤ N ≤ 200, 1 ≤ M ≤ 10^9。
题解：

$Code：$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define N 205
using namespace std;
int n,m;
double ans,p[N],f[N][2];
int main()
{
    scanf("%d%d",&n,&m);
    p[1]=1;
    for(int i=2;i<=n;i++)p[i]=p[i-1]/m;
    f[0][1]=1;
    for(int i=0;i<=n;i++)
        for(int j=i+1;j<=n;j++)
        {
            f[j][0]+=(j-i)*p[j-i]*(f[i][0]*(m-2)/m+f[i][1]*(m-1)/m);
            f[j][1]+=(j-i)*p[j-i]*f[i][0]*1/m;
        }
    ans=p[n]*n;
    for(int i=1;i<n;i++)
        ans+=i*i*f[n-i][0]*p[i];
    printf("%.5lf",ans);
    return 0;
}