BZOJ 2554: Color【条件概率】

题目描述：

有n个球排成一列，每个球都有一个颜色，用A-Z的大写字母来表示，我们每次随机选出两个球ball1,ball2,使得后者染上前者的颜色，求期望操作多少次，才能使得所有球的颜色都一样？
输出保留一位小数。

题目分析：

liu_runda大佬的博客

把答案分成每种颜色，就是求最后合成某种颜色的概率*合成这种颜色的期望步数。

用$g[i]$表示这种颜色有$i$个，最后$n$个全部变成这种颜色的概率。
容易得到$g[i]=\frac{1}{2}g[i-1]+\frac{1}{2}g[i+1]$，边界$g[0]=0,g[n]=1$
可以看出$g[i]=\frac{i}{n}$

用$f[i]$表示这种颜色有$i$个，最后$n$个全部变成这种颜色的期望步数。
总共有$n\ast (n-1)$种方案，会使这种颜色数量改变的方案有$2\ast i\ast (n-i)$种，所以数量改变的概率P可求，期望$\frac{1}{P}$步后数量会改变。
但是注意，我们是在最终全部都会变成这种颜色的条件下算的期望步数，所以我们的转移需要去掉那些不能到达这个条件的概率，$i$转移到$i-1$和$i+1$的概率分别是$\frac{1}{2}$，由$i-1$到达最终状态的概率是$g[i-1]$，所以$f[i-1]$对$f[i]$的贡献是$\frac{1}{2}\ast g[i-1]$，而不是简单的$\frac{1}{2}$。同理，$f[i+1]$的贡献是$\frac{1}{2}\ast g[i+1]$。令$G=\frac{1}{2}\ast g[i-1]+\frac{1}{2}\ast g[i+1]$，就有：$$f[i]=\frac{1}{G}\left(\frac{1}{2}\ast g[i-1]\ast f[i-1]+\frac{1}{2}\ast g[i+1]\ast f[i+1]\right)+\frac{1}{P}$$
化简一下就是$$f[i]=(i-1)/(2i)\ast f[i-1]+(i+1)/(2i)\ast f[i+1]+\frac{n\ast (n-1)}{2\ast i\ast (n-i)}$$
$f[0]$无意义，$f[n]=0$，把$f[i]$代成$a[i]\ast f[1]+b[i]$的形式递推，算出$f[n]=a[n]\ast f[1]+b[n]$，解出$f[1]$即可。

Code：

#include<bits/stdc++.h>
#define maxn 10005
using namespace std;
int n,cnt[26];
double f1,a[maxn],b[maxn],ans;
char s[maxn];
int main()
{
	scanf("%s",s),n=strlen(s);
	for(int i=0;i<n;i++) cnt[s[i]-'A']++;
	a[1]=1,b[1]=0;
	for(int i=1;i<n;i++){
		a[i+1]=(a[i]-a[i-1]*(i-1)/(2*i))*(2*i)/(i+1);
		b[i+1]=(b[i]-b[i-1]*(i-1)/(2*i)-1.0*n*(n-1)/(2*i*(n-i)))*(2*i)/(i+1);
	}
	f1=-b[n]/a[n];
	for(int i=0;i<26;i++) ans+=1.0*cnt[i]/n*(a[cnt[i]]*f1+b[cnt[i]]);
	printf("%.1f\n",ans);
}