codeforces 9D How many trees(动态规划)

题目链接: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;
}

 

上一篇:报错Too many connections


下一篇:web之django的序列化器汇总