（纪中）1930. Colorful Circle【期望DP】

---

解题思路

设$dp[i][1/0]$表示到i位置，当前位置与首尾是否相同，相同为$1$，不同为$0$。
$p[i]$表示i个宝石颜色相同的概率。

枚举当前点i与i之前第一个与它异色的点j，则j+1~i的颜色都相同转移方程为：

• $dp[i][0]+=dp[j][0]\ast p[i-j]\ast (i-j)\ast (M-2.0)+dp[j][1]\ast p[i-j]\ast (i-j)\ast (M-1.0);$
• $dp[i][1]+=dp[j][0]\ast p[i-j]\ast (i-j);$

$m-2$因为当前的颜色不仅与j不一样，与首尾也不一样，$m-1$因为与$j$不一样，$i-j$ 是这段的长度的贡献，p[i-j]为i-j个宝石颜色相同的概率，$i-j$为获得的价值.
第二个方程不乘可行的颜色数是因为i后面可能还有与i颜色相同的点，到时候会在算$d[][0]$中统计

求答案的话就从首尾段来处理,首先很显然的 $ans=Pn\ast n\ast M$（整个环同色），那么再考虑别的情况对答案的贡献:枚举首尾段的总长度i：

• $ans+=i\ast i\ast dp[n-i][0]\ast p[i]\ast M;$

(一个i是价值，一个i是分割头尾的方式数，也就是切割方案,比如两个点那就可以$n-2$到$n$或是$n-1$到$1$，首尾段所有颜色都可以选，共M种，p[i]表示首尾段所有颜色相同的概率）

---

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iomanip>
#include<cmath>
#define ldb long double
using namespace std;

const ldb e=1e-18;
int n,m;
ldb p[210],ans,dp[210][5];

int main(){
	scanf("%d%d",&n,&m);
	ldb ppp=(1*1.000000)/(m*1.000000);
	p[0]=1;
	for(int i=1;i<=n;i++)
		p[i]=p[i-1]*ppp;
	ldb M=1.0*m;;
	dp[0][0]=0;dp[0][1]=1;
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<i;j++)
		{
			dp[i][0]+=dp[j][0]*p[i-j]*(i-j)*(M-2.0)+dp[j][1]*p[i-j]*(i-j)*(M-1.0);
			dp[i][1]+=dp[j][0]*p[i-j]*(i-j);
		}
	}
	ans=p[n]*n*M;
	for(int i=1;i<n;i++)
		ans+=i*i*dp[n-i][0]*p[i]*M;
	printf("%.10Lf",ans);
}