【51nod 1412】AVL树的种类 (dp)

题意

平衡二叉树(AVL树），是指左右子树高度差至多为1的二叉树，并且该树的左右两个子树也均为AVL树。
求 $n$ 个点的AVL树有多少种形态

题解

设 $f[i][j]$ 表示结点数为 $i$，高度为 $j$ 的种类数
$f[i][j]=\sum f[k][j-1]*f[i-1-k][j-1]+f[k][j-1]*f[i-1-k][j-2]+f[k][j-2]*f[i-1-k][j-1]$

代码

#include <cstdio>
#include <cstring>
#define N 2010
#define mod 1000000007
int n;
long long ans=0,f[N][25];
int main(){
    scanf("%d",&n);
    f[0][0]=f[1][1]=1;
    for(int x=2;x<=n;x++){
        for(int y=2;y<20;y++){
            for(int k=0;k<x;k++){
                f[x][y]=(f[x][y]+f[k][y-1]*f[x-1-k][y-1]%mod+f[k][y-1]*f[x-1-k][y-2]%mod+f[k][y-2]*f[x-1-k][y-1]%mod)%mod; 
            }
        }
    }
    for(int i=0;i<20;i++) ans=(ans+f[n][i])%mod;
    printf("%lld\n",ans);
    return 0;
}