[代码元]平衡二叉树

题目描述
平衡二叉树(AVL树），是指左右子树高度差至多为1的二叉树，并且该树的左右两个子树也均为AVL树。 现在问题来了，给定AVL树的节点个数n，求有多少种形态的AVL树恰好有n个节点。
输入描述
一行一个整数T(T≤2000)表示数据组数
T行, 每行一个整数n(n≤2000)
输出描述
每行一个数字表示结果，由于结果巨大，输出它对109+7取余数的结果。
输入样例
1
10
输出样例
60

分析：

一棵树是平衡二叉树，其左子树有子树满足同样性质。
性质：左右子树的高度差最多为1
则我们可以推导出树与他的左右子树的关系。
设dp[i][j]表示有i个节点，高度为j时候的方案数。
显然当0个结点时，算一个方案，只有一个节点时也算一个方案。
即dp[0][0] = dp[1][1] = 1
状态转移方程课表示为如下：
左右子树同样高：dp[i][j] = dp[L][j - 1] + dp[R][j - 1]
左子树高右子树低：dp[i][j] = dp[L][j - 1] + dp[R][j - 2]
左子树低右子树高：dp[i][j] = dp[L][j - 2] + dp[R][j - 1]
其中节点的关系应该为 i = L + R + 1

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 2010, mod = 1e9 + 7;

int n;
int dp[N][N];

int main()
{
	dp[0][0] = dp[1][1] = 1;
	for (int i = 2; i <= 2000; i ++ )
	{
		for (int j = 2; j <= 20; j ++ )
		{
			for (int l = 0; l < i; l ++ )
			{
				int r = i - 1 - l;
				dp[i][j] = (dp[i][j] + (long long)dp[l][j - 1] * dp[r][j - 1] % mod) % mod;
				dp[i][j] = (dp[i][j] + (long long)dp[l][j - 2] * dp[r][j - 1] % mod) % mod;
				dp[i][j] = (dp[i][j] + (long long)dp[l][j - 1] * dp[r][j - 2] % mod) % mod;
			}
		}
	}

	// for (int i = 0; i <= 20; i ++ )
	// {
	// 	for (int j = 0; j <= 15; j ++ )
	// 		printf("cnt = %d, h = %d, val = %d\n", i, j,dp[i][j]);
	// }

	int t;
	cin >> t;
	while (t -- )
	{
		scanf("%d", &n);

		int ans = 0;
		for (int i = 0; i <= 20; i ++ )
			ans = (ans + dp[n][i]) % mod;

		printf("%d\n", ans);
	}
	return 0;
}