题目链接:https://vjudge.net/problem/CodeForces-9D
题目大意
给你一个n,h,问你n个节点能够构成深度不小于h的二叉搜索树的个数是多少个?
思路:
很明显的树型dp,dp[n][h]表示n个节点深度小于等于h,此时答案为 dp[n][n] - dp[n][h-1] , 状态转移条件是两个儿子深度都小于等于h-1 , 即
dp[i][j]=sum{dp[k][j-1]*dp[i-k-1][j]},表示左右子树深度都小于h-1时,使用乘法计数原则
只有一种情况,每次转移需要O(n)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <queue>
#include <cmath>
#include <vector>
#include <iomanip>
#include <map>
#include <set>
#define int long long
using namespace std;
const int maxn = 40;
int dp[maxn][maxn];
void init()
{
memset(dp, 0, sizeof(dp));
for (int i = 0; i <= 35; i++) dp[0][i] = 1;
for (int i = 1; i <= 35; i++)
{
for (int j = 1; j <= 35; j++)
{
for (int k = 0; k < i; k++)
{
dp[i][j] += (dp[k][j - 1] * dp[i - k - 1][j - 1]);
}
}
}
}
signed main() {
init();
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
int n, k;
cin >> n >> k;
cout << dp[n][n] - dp[n][k-1] << endl;
return 0;
}