【51nod】1412 AVL树的种类

题目：

点击打开链接

动态方程：

令dp[i][k]为当前总节点数i，最大深度为k的AVL的方案数，那么有两种转移，一种是当前的AVL左右子树的深度一样，或者左右子树的深度差为1

dp[i][k]+=dp[i-1-j][k-1]*dp[j][k-1]
我们枚举 j为右子树的总结点数

dp[i][k]+=2*dp[i-1-j][k-2]*dp[j][k-1]
这种是左右子树深度不一样的情况，有可能左边少1，或者右边少1，但都是等价的，所以直接*2就可以了

代码：

//把h理解成层数才是2^h-1等于结点数;
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int MOD=1e9+7;
const int MAXN=2005;

ll dp[MAXN][16];

int main()
{
    int n;
    cin>>n;
    dp[0][0]=1;dp[1][1]=1;
    //i为结点数，k为层数，j为右子数结点数
    for(int i=2;i<=n;i++)
    {
        for(int k=2;k<16;k++)
        {
            for(int j=0;j<i;j++)
            {
                (dp[i][k]+=(dp[i-1-j][k-1]*dp[j][k-1])%MOD)%=MOD;
                (dp[i][k]+=2*(dp[i-1-j][k-1]*dp[j][k-2])%MOD)%=MOD;
            }
        }
    }
    ll count=0;
    for(int k=1;k<16;k++)
    {
        count+=dp[n][k];
        count%=MOD;
    }
    printf("%lld\n",count);
    return 0;
}