【SHOI2012】BZOJ2830随机树题解（概率期望+DP）

题目：BZOJ2830.
题目大意：一棵有$n$个叶子的二叉树可以通过$n-1$次等概率随机往一个叶节点上加两个儿子生成.现在给定$q,n$，当$q=1$时输出$n$个叶子的二叉树叶子深度平均值的期望，$q=2$输出树高的期望，注意根节点深度为$0$.
$1\le q\le 2,1\le n\le 100$.

考虑$q=1$，很容易想到设$f[i]$表示$i$个叶子的答案，方程：
$$f[i]=\frac{f[i-1]\ast (i-1)+f[i-1]+2}{i}=f[i-1]+\frac{i}{2}$$

考虑$q=2$，发现只设一维状态不行了，考虑加一维状态，设$f[i][j]$表示$i$个叶子深度为$j$的概率.通过枚举左子树的叶子数，我们可以列出如下方程：
$$f[i][j]=\sum _{k=1}^{i-1}p[i][j](f[k][j-1]\sum _{t=0}^{j}f[i-k][t-1]+f[i-k][j-1]\sum _{t=0}^{j}f[k][t]-f[k][t]\ast f[i-k][t])$$

其中$p[i][j]$表示$i$个叶子的树$j$个在左子树的概率.

我们发现只要能够预处理出$p[i][j]$就可以$O(n^4)$解决这个问题了，可是这个不够快.

考虑这两个式子中都有一个和的形式，可以考虑前缀和，但是由于计算答案的式子：
$$E(X)=\sum _{i=1}^{n}f[n][i]\ast i$$

用前缀和来处理这个式子有些麻烦，所以考虑变成后缀和，即$f[i][j]$表示$i$个叶子深度至少为$j$的概率，那么：
$$\begin{gathered} E(X)=\sum _{i=1}^{n}f[n][i] \\ f[i][j]=\sum _{k=1}^{i-1}p[i][j](f[k][j-1]+f[i-k][j-1]-f[k][j-1]\ast f[i-k][j-1]) \end{gathered}$$

这个时候计算的复杂度就变成$O(n^3)$的了，但是我们还不知道$p[i][j]$呢.

其实$p[i][j]=\frac{1}{i-1}$，这个东西用数学归纳法证明：
$$p[i][j]=p[i-1][j-1]\ast \frac{j-1}{i-1}+p[i-1][j]\ast \frac{i-1-j}{i-1}$$

若$p[i-1][j]=\frac{1}{i-2}$，那么：
$$p[i][j]=\frac{1}{i-2}(\frac{j-1}{i-1}+\frac{i-1-j}{i-1})=\frac{1}{i-1}$$

那么代码如下：

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

#define Abigail inline void
typedef long long LL;

const int N=100;

int n,q;
double f1[N+9],f2[N+9][N+9],ans;

Abigail into(){
  scanf("%d%d",&q,&n);
}

Abigail work1(){
  f1[1]=0;
  for (int i=2;i<=n;++i)
    f1[i]=f1[i-1]+2.0/i;
  ans=f1[n];
}

Abigail work2(){
  for (int i=1;i<=n;++i) f2[i][0]=1;
  for (int i=2;i<=n;++i)
    for (int j=1;j<i;++j){
      for (int k=1;k<i;++k)
        f2[i][j]+=f2[k][j-1]+f2[i-k][j-1]-f2[k][j-1]*f2[i-k][j-1];
      f2[i][j]/=i-1;
	}
  for (int i=1;i<n;++i)
    ans+=f2[n][i];
}

Abigail outo(){
  printf("%.6lf\n",ans);
}

int main(){
  into();
  q==1?work1():work2();
  outo();
  return 0;
}