Codeforces9 D How many trees?（二叉树相关，计数dp）

题意：

给定n和h，问n个点构成的二叉树有多少种高度>=h的

数据范围：n<=35，h<=n

解法：

这题要么是组合数学,要么是dp,
第一眼看起来更像dp,(组合数学想不到怎么解)

考虑如何拆解为重叠子问题,
错误思路:
我一开始想的是每种状态记录:无子节点,1子节点,2子节点的点数量,然后d[n]当作往d[n-1]插入一个点.
发现好像转移不来,然后凉了.

一种正确思路:
可以把新加入的一个点看成根,连接左右两颗子树,那么只需要枚举左右两棵子树的状态就行了.
令d[i][j]表示i个点,高度为j的方案数
枚举左子树的节点个数l,那么可以O(1)计算出右子树的节点个数r.
再枚举左子树高度a和右子树高度b,
转移方程:d[i][max(a,b)+1]+=d[l][a]*d[r][b]

i一层枚举,l,a,b三层,一共四层,总复杂度O(n^4)

总结:二叉树类计数有左右子树合并的计数方法

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int A=40;
int d[A][A];
int n,h;
signed main(){
    cin>>n>>h;
    d[0][0]=1;
    for(int i=1;i<=n;i++){
        for(int l=0;l<i;l++){//枚举左子树节点数
            int r=i-l-1;//计算出右子树节点数,多减的1是根节点
            for(int a=0;a<=l;a++){//枚举左子树深度
                for(int b=0;b<=r;b++){//枚举右子树深度
                    d[i][max(a,b)+1]+=d[l][a]*d[r][b];
                }
            }
        }
    }
    int ans=0;
    for(int i=h;i<=n;i++){
        ans+=d[n][i];
    }
    cout<<ans<<endl;
    return 0;
}

---