51nod 1412 AVL树的种类(dp)

题目链接：51nod 1412 AVL树的种类

开始做的时候把深度开得过小了结果一直WA，是我天真了。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int mod = 1e9+7;
const int N = 2001;
ll dp[N][16];//dp[i][k]  i:节点数, k:深度
void solve(){
    int i, j, k;
    dp[0][0] = dp[1][1] = 1;
    for(i = 2; i < N; ++i){
        for(k = 2; k < 16; ++k){
            for(j = 0; j < i; ++j){
                //j为右子树的节点数,左右子树节点之和为i-1
                //左右子树深度相同
                dp[i][k] += dp[i-1-j][k-1] * dp[j][k-1];
                //左右子树深度差一
                dp[i][k] += 2 * dp[i-1-j][k-2] * dp[j][k-1];
                dp[i][k] %= mod;
            }
        }
    }
}
int main(){
    int n, i;
    ll ans = 0;
    solve();
    scanf("%d", &n);
    for(i = 1; i < 16; ++i){
        ans += dp[n][i];
    }
    ans %= mod;
    printf("%lld
", ans);
    return 0;
}