暑期个人赛题目 CF赛题 How many trees?
题目链接 http://codeforces.com/problemset/problem/9/D
一开始看到这道题还认为世道数据结构,比完之后才发现这是一道动态规划的题目,啊啊啊,是我太菜
虽然这道题左学长用了一个四维的DP,但其实只需要3个循环加一个二维DP数组即可
dp [ i ][ j ]表示i个点在高度为j的情况下所得的组合数
但是对于每一个不符合题目要求的(就比如7个点高度只有2),这种不可能的情况,我们都要赋值成1,因为最后的大难都是左右子树的组合数想乘得来的,并且左右子树互不影响。
影响左右子树的只有在这棵树上面的根节点的编号,若根节点为k,则左子树的节点数为k-1,右子树为n-k
转移方程就是 dp[i][j]+=dp[k-1][j-1]*dp[i-k][j-1]
上代码
#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n,h;
ll dp[40][40];//dp[i][j]代表i个点形成高度为j的树的个数
int main(){
scanf("%d %d",&n,&h);
memset(dp,0,sizeof(dp));
for(int i=0;i<=n;i++)
dp[0][i]=1;
for(int j=1;j<=n;j++)//对每个高度进行遍历
for(int i=1;i<=n;i++)//对每个高度里的左右子节点下的点的数量遍历
for(int k=1;k<=i;k++)//对每个高度下做根节点的节点编号进行遍历
dp[i][j]+=dp[k-1][j-1]*dp[i-k][j-1];
printf("%lld",dp[n][n]-dp[n][h-1]);
return 0;
}