[SHOI2012]随机树 洛谷P3803 【期望dp】

题面

---

SOL

分两个问题讨论；
Q1 ：显然的期望dp定义式 ：$f[i]:第i步的期望平均叶子深度$,
转移方程： $\frac{f[i]=f[i-1]\ast (i-1)+f[i-1]+2}{i},->f[i]=f[i-1]+\frac{2}{i}.$
可以再数学化简，这里没必要了。

Q2: 上一问好做，是因为我们可以直接从任意一个点转移。这一问要求最大深度的期望，如果要这样做，还要维护一个最大深度对应的叶子数。留作讨论吧。 。(updated, 9.23 ,这样定义状态包含了条件本身的概率，而子问题的条件和当前的条件不一样，不知道条件之间的转移概率，所以这个方法不行。）

从"拼树"的角度思考，一颗树由 一个根节点加左右两个子树拼接而成，左右子树是原问题的子问题，于是我们定义$f[i][j]表示有i个叶子节点的树，深度为j的概率$，

那么枚举左右两子树，加上当前根节点就可以直接转移了。（注意本题特殊性，还要乘以生成左右子树叶节点树对应的概率）

法一：

$$f[i][j]=\sum _{L=1}^{i-1}p[i][L]\ast (\sum _{k=1}^{j-1}f[L][j-1]\ast f[i-L][k]+\sum _{k=1}^{j-1}f[L][k]\ast f[i-L][j-1]-f[L][j-1]\ast f[i-L][j-1])$$
$$p[i][j]=p[i-1][j-1]\ast \frac{j-1}{i-1}+p[i-][j]\ast (1-\frac{j}{i-1})$$
这里$f$数组可以前缀和转移优化，$O(n^3)$统计答案：$ans={\sum }_{j=1}^{n-1}f[n][j]\ast j.$

法二

使用全期望公式：
$f[i][j]:i个叶节点的子树，深度大于等于j的概率$，
$ans={\sum }_{i=1}^{n-1}f[n][i]$.
转移有：
$$f[i][j]=\sum _{L=1}^{i-1}p[i][L]\ast (f[L][j-1]+f[i-L][j-1]-f[L][j-1]\ast f[i-L][j-1])$$
一样 O(n^3)解决 （左右子树都大于等于 $j-1$的概率被算了两次，减掉即可）

PS

$p[i][k=1...i-1]$ 其实就等于 $\frac{1}{i-1}$。
证明：
先选左右子树,生成选法序列，再在子树里选叶节点,生成选树数列。

对于$p[i][j]$，
从1个叶子节点开始构造，每次选L，L增加一个，选R，R增加一个；
总共选了 k 个 L ，选的方案是 $\left(\genfrac{}{}{0ex}{}{i-2}{k}\right)$，为$\frac{(i-2)!}{(i-2-k)!\ast k!}$.

对于子树的拼接方案，左子树叶节点从1 到 $k$，总共$k!$总生成方法。同理，右子树总共$(i-2-k)!$种生成方法。
所以总的选法是序列生成方案*子树生成方案，为$\frac{(i-2)!}{(i-2-k)!\ast k!}\ast k!\ast (i-2-k)!=(i-2)!$

所以总方案数和左右儿子个数无关，$p[i][j]=\frac{1}{i-1}$.
证毕。

---

CODE

#include<bits/stdc++.h>
#define pf printf
#define sf scanf
#define cs const
#define ll long long
#define db double
#define ri register int
using namespace std;
cs int N=110;
db p[N][N],f[N][N],g[N];
int n;
inline void init_p(){
	p[2][1]=1;
	for(ri i=3;i<=n;++i){
		for(ri j=1;j<i;++j){
			p[i][j]=p[i-1][j-1]*(j-1)/(1.0*(i-1))+p[i-1][j]*(1.0-1.0*j/(1.0*(i-1)));
		}
	}
}
inline void init_f(){
	f[1][0]=1;
	for(ri i=2;i<=n;++i){
		f[i][1]=1;
		for(ri j=2;j<i;++j){
			for(ri k=1;k<i;++k){
				f[i][j]+=p[i][k]*(f[k][j-1]+f[i-k][j-1]-f[k][j-1]*f[i-k][j-1]);
			}
		}
	}
	db ans=0;
	for(ri i=1;i<n;++i)ans+=f[n][i];
	printf("%.6lf",ans);
}
inline void init_g(){
	g[1]=0;
	for(ri i=2;i<=n;++i)g[i]=g[i-1]+2.0/(1.0*i);
	printf("%.6lf",g[n]);
}
signed main (){
	int q;sf("%d%d",&q,&n);
	if(q==1)init_g();
	if(q==2)init_p(),init_f();
	return 0;
}