Problem M: qwb与二叉树
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 179 Solved: 45
[ Submit][ Status][ Web Board]
Description
某一天,qwb正在上数据结构课。老师在讲台上面讲着二叉树,qwb在下面发着呆。
突然qwb想到一个问题:对于一棵n个无编号节点,m个叶子的有根二叉树,有多少种形态呐?你能告诉他吗?
Input
多组输入,处理到文件结束,大约有104组数据。
每一组输入一行,两个正整数n,m(0≤m≤n≤50),意义如题目所述。
Output
每一行输出一个数,表示相应询问的答案,由于答案可能很大,请将答案对109+7取模后输出。
Sample Input
4 2
10 5
Sample Output
6
252
HINT
样例1的6种形态:
正解:点击打开链接
我是在上面博客里学到的,卡特兰数竟然还能表示n个节点的二叉树形态数。。真的神奇。。
总结:先说n个节点的二叉树有多少种形态的求法,用到的思想是递归+分治,先确定一个点为根,求左子树的形态数*右子树的形态数即为该点为根时的总形态数,而左子树和右子树的形态数又可以转化为同样的问题来解决。本题不过是加上了一个有多少叶节点的限制,也很简单,再嵌套一层for循环枚举左右子树各有多少叶节点就行了。
求解过程大部分是重复的,因此可以加上记忆化的步骤,dp[i][j]表示i个节点j个叶节点的二叉树有多少种形态。
代码:
#include<bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
const int mod=1e9+7;
ll dp[56][56];
ll dfs(int n,int m)
{
if(dp[n][m]!=-1)return dp[n][m];
dp[n][m]=0;
for(int i=0;i<n;i++)
for(int j=0;j<=i&&j<=m;j++)
dp[n][m]+=dfs(i,j)*dfs(n-1-i,m-j),dp[n][m]%=mod;
return dp[n][m]%mod;
}
int main()
{
int n,m;
memset(dp,-1,sizeof(dp));
dp[0][0]=dp[1][1]=1;
dp[1][0]=0;
dfs(51,51);//注意这里一定要大于50
while(~scanf("%d%d",&n,&m))
printf("%lld\n",dp[n][m]);
return 0;
}