CodeForces 9 D.How many trees?（dp）

Description

求$n$个节点构成的高度不小于$h$的二叉树数量

Input

两个整数$n,h(1\le h\le n\le 35)$

Output

输出$n$个节点构成的高度不小于$h$的二叉树数量

Sample Input

3 2

Sample Output

5

Solution

$dp[n][h]$表示$n$个节点构成的高度为$h$的二叉树数量，枚举左儿子节点数量$x$和高度$h_1$以及右儿子高度$h_2$，则右儿子节点数量为$n-1-x$，进而有转移方程$dp[n][h]=\sum\limits_{h_1,h_2,x}dp[x][h_1]*dp[n-1-x][h_2]$，记忆化一下即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>P;
const int maxn=40;
bool flag[maxn][maxn];
ull dp[maxn][maxn];
ull Solve(int n,int h)
{
    if(flag[n][h])return dp[n][h];
    flag[n][h]=1;
    if(n==h)
    {
        if(n==0)return dp[n][h]=1;
        return dp[n][h]=1ull<<(n-1);
    }
    if(((1ll<<h)-1<n)||n<h)return dp[n][h]=0;
    ull ans=0;
    for(int h1=0;h1<h;h1++)
        for(int h2=0;h2<h;h2++)
            if(max(h1,h2)==h-1)
                for(int x=0;x<n;x++)
                    ans+=Solve(x,h1)*Solve(n-1-x,h2);
    return dp[n][h]=ans;
}
int main()
{
    int n,h;
    while(~scanf("%d%d",&n,&h))
    {
        ull ans=0;
        for(int i=h;i<=n;i++)ans+=Solve(n,i);
        printf("%I64u\n",ans);
    }
    return 0;
}