洛谷3830 BZOJ2830 SHOI2012 随机树 概率期望 dp 组合数学

题目链接

题意：
有两问，第一问是所有可能的$n$个叶子节点的二叉树的叶节点的平均深度，第二问是$n$个叶节点的二叉树的期望深度。$n<=100$，结果保留$6$位小数。

题解：
我们一问一问的来做。

首先考虑第一问。我们设$f[i]$表示有$i$个叶子的二叉树的平均深度。我们考虑把一个叶子节点展开之后的影响。我们的想法是，用总深度除以叶子数。我们发现我们可以算出新增加的叶子的平均深度，我们把原来的某一个叶子变成一个非叶子节点，然后增加了两个深度是原来叶子节点深度$+1$的节点，平均来讲，变化量就是$-f[i]+2\ast (f[i]+1)=f[i]+2$。我们用原来的平均总深度加上多了一个叶子后的平均深度增加量除以叶子总数就应该是新的平均总深度，列出式子就是$f[i]=\frac{f[i-1]\ast (i-1)+f[i]+2}{i}=\frac{f[i-1]\ast i+2}{i}=f[i-1]+\frac{2}{i}$

这样就可以$O(n)$的递推一下得到最终结果了。

然后考虑第二问。这个第二问就比较有难度了。我们一个简单的想法是设$dp[i][j]$表示考虑了$i$个叶子，最大深度是$j$的概率，这样最终答案就是${\sum }_{i=1}^{n-1}dp[n][i]\ast i$。但是我们发现这个式子不太好转移。

有一个巧妙的式子是，我们设$dp[i][j]$表示现在有$i$个叶子，最大深度大于等于$j$的概率。这样设的话，答案就是${\sum }_{i=1}^{n-1}dp[n][i]$，因为深度为$x$的大于等于$1-x$的每一个，会在$dp[n][1]$到$dp[n][x]$都被统计进去一遍，于是会被算正好$x$遍，就不用容斥也不用乘权值了。

我们考虑这个式子怎么转移。我们要从最大深度大于等于$j-1$的转移到最大深度大于等于$j$的。我们枚举左子树有多少个叶子，设为$k$，这样我们可以算出右子树的叶子，只要之前那么多叶子的左右子树有一个深度大于等于$j-1$，那么原树就算是深度大于等于$j-1$。我们用两者的概率相加，发现都大于等于$j-1$的部分被算了两遍，要减去一遍。那么我们用这个概率乘$i$个叶子，左子树有$k$个叶子的概率，设为$p$，最后累加一下就是答案。列出式子就是$dp[i][j]={\sum }_{k=1}^{i-1}p\ast (dp[k][j-1]+dp[i-k][j-1]-dp[k][j-1]\ast dp[i-k][j-1])$

现在的问题是求出$p$，只要能求出$p$，我们就可以在$O(n^3)$的时间里求出答案。我们考虑第一次操作一定是展开根，变成左右子树都有一个叶子的情况。每一次操作会使一个叶子不再是叶子，并产生两个新的叶子，于是每次操作叶子数会加一，于是$i$个叶子的树是经过了$i-1$次操作，除掉第一次对根的操作外，每次操作的要么是左子树，要么是右子树，共$i-2$次。我们考虑用要求的情况的方案数除以总的方案数来表示这个概率。我们在这$i-2$次操作中操作了左子树$k-1$次，并且这$k-1$次操作可以形成不同的形态的树。对于$i-2$次操作了左子树$k-1$次，方案数就是$C_{i-2}^{k-1}$。我们考虑形成$i$个叶子的树的方案数（这里形成的树的形态可能相同）。我们要展开$n-1$次，我们第一次只能展开根，第二次有两个不同的叶子可以选，第三次有三个叶子可以选，第$n-1$次有$n-1$个叶子可以选，于是方案数应该是$(n-1)!$种。那么我们就能列出式子了：$$p=\frac{C_{i-2}^{k-1}\ast (k-1)!\ast (i-k-1)!}{(i-1)!}$$ $$=\frac{\frac{(i-2)!}{(k-1)!\ast ((i-2)-(k-1))!}\ast (k-1)!\ast (i-k-1)!}{(i-1)!}$$ $$=\frac{\frac{(i-2)!}{(k-1)!\ast (i-k-1)!}\ast (k-1)!\ast (i-k-1)!}{(i-1)!}$$ $$=\frac{(i-2)!}{(i-1)!}=\frac{1}{i-1}$$ 于是我们推出最后的概率与$k$无关，始终是$\frac{1}{i-1}$

那么我们带回原式，就得到是dp式子：$dp[i][j]={\sum }_{k=1}^{i-1}\frac{dp[k][j-1]+dp[i-k][j-1]-dp[k][j-1]\ast dp[i-k][j-1]}{i-1}$

这样就做完第二问了。复杂度是$O(n^3)$。

代码：

#include <bits/stdc++.h>
using namespace std;

int n,q;
double dp[110][110],f[110],ans;
int main()
{
	scanf("%d%d",&q,&n);
	if(q==1)
	{
		f[1]=0;
		for(int i=2;i<=n;++i)
		f[i]=f[i-1]+2.0/(double)i;
		printf("%.6lf\n",f[n]);
	}
	else
	{
		for(int i=1;i<=n;++i)
		dp[i][0]=1;
		for(int i=2;i<=n;++i)
		{
			for(int j=1;j<=i-1;++j)
			{
				for(int k=1;k<=i-1;++k)
				dp[i][j]+=1.0/(double)(i-1)*(dp[k][j-1]+dp[i-k][j-1]-dp[k][j-1]*dp[i-k][j-1]);
			}
		}
		for(int i=1;i<=n-1;++i)
		ans+=dp[n][i];
		printf("%.6lf\n",ans);
	}
	return 0;
}