USACO 2.3 nocows

 

这题是比较明显的用动态规划计数的问题，不过状态转移方程不是很简单（或许是我没想到简单的）。

用n（结点数）和k（高度）表示状态。

f[n,k]=2*sum{f[n-1-i,k-1]*g[i,k-1]} - sum{f[n-1-i,k-1]*f[i,k-1]} for all 0<i<n-1（因为左右子树都不为空）

n=1,k=1是边界情况。

这个关系式的依据是：

由于左右子树都不为空，于是左子树或者右子树高度为k-1。左右对称，于是只枚举左子树高度k-1的情况之后乘以2就行了。

但是左右子树高度同时为k-1的情况数了2遍，所以再减掉一次。思路和容斥原理一样。

 

g[n,k]表示结点数为n，高度不超过k的树的总数。所以

g[n,k]=sum{f[n,i]} for all 0<i<=k

 

 

来自nocow的更漂亮的解法：

首先明确一下题目的意思：用N个点组成一棵深度为K的二叉树，求一共有几种方法？ 设dp[i,j]表示用i个点组成深度最多为j的二叉树的方法数，则：

dp[i,j]=∑(dp[k,j-1]×dp[i-1-k,j-1])(k∈{1..i-2})
边界条件：dp[1,i]=1

我们要求的是深度恰好为K的方法数S，易知S=dp[n,k]-dp[n,k-1]。 但需要注意的是，如果每次都取模，最后可能会有dp[n,k]<dp[n,k-1],所以可以用S=(dp[n,k]-dp[n,k-1]+v) mod v