51nod BSG白山极客挑战赛 AVL树的种类

                                                                                          AVL树的种类

平衡二叉树(AVL树），是指左右子树高度差至多为1的二叉树，并且该树的左右两个子树也均为AVL树。 现在问题来了，给定AVL树的节点个数n，求有多少种形态的AVL树恰好有n个节点。

Input

一行，包含一个整数n。 (0 < n <= 2000)

Output

一行表示结果，由于结果巨大，输出它对1000000007取余数的结果。

Input示例

10

Output示例

60

     思路：经典dp ，dp[i][h]=dp[a][h-1]*dp[i-a-1][h-1]+dp[a][h-2]*dp[i-a-1][h-1]+dp[a][h-1]*dp[i-1-a][h-2] ，dp[i][j]表示i个结点构成高度为j的AVL数的方法数。一个结点高度按1来算，不然后面不好写了。

代码时间到：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 1e4+100;
const int MOD=1e9+7;
#define LL long long 
LL dp[maxn][20];
//dp[i][h]=dp[a][h-1]*dp[i-a-1][h-1]+dp[a][h-2]*dp[i-a-1][h-1]+dp[a][h-1]*dp[i-1-a][h-2]
int main(){
    int n;
    scanf("%d",&n);
    int i,j,k;
    memset(dp,0,sizeof(dp));
    dp[1][1]=1;
    dp[0][0]=1;
    for(i=2;i<=n;i++){
        for(j=0;j<i;j++){
            for(k=1;k<=20;k++){
                int d=i-j-1;
                dp[i][k]+=dp[j][k-1]*dp[d][k-1];
                dp[i][k]%=MOD;
                dp[i][k]+=2*dp[j][k-2]*dp[d][k-1]%MOD;
                dp[i][k]%=MOD;
                
            }
        }
    }
    int ans=0;
    for(i=1;i<=n;i++) ans=(ans+dp[n][i])%MOD;
    cout<<ans<<endl;
    return 0;
}