[51nod 1412]AVL树的种类

题目大意

求有多少种节点个数为n的AVL树。

吼简单

求有多少个节点为n的树，容易想到f[i]表示用i个节点弄出一颗树。
而AVL有神奇性质，我们要设f[i,j]表示深度为ij个节点。
那我不是n^3？
AVL学好了，你容易知道AVL的深度是log n的。
所以是n^2 log n的复杂度。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=2000+10,mo=1000000007;
int f[25][maxn];
int i,j,k,l,t,n,m,ans;
int main(){
    f[0][0]=1;
    f[1][1]=1;
    fo(i,2,15){
        fo(j,1,2000)
            fo(k,0,j-1){
                f[i][j]=(f[i][j]+(ll)f[i-1][k]*f[i-1][j-k-1]%mo)%mo;
                if (i>1) f[i][j]=(f[i][j]+(ll)f[i-1][k]*f[i-2][j-k-1]%mo*2%mo)%mo;
            }
    }
    while (scanf("%d",&n)!=EOF){
        ans=0;
        fo(i,1,15) ans=(ans+f[i][n])%mo;
        printf("%d\n",ans);
    }
}