Given n, how many structurally unique BST's (binary search trees) that store values 1...n?
For example,
Given n = 3, there are a total of 5 unique BST's.
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
这道题的意思就是给出一个数n即一共有1-n 共 n个数,然后问可以用多少种不同的二叉树来存储这些数据。(这些二叉树都可以满足中序遍历的条件下按1-n的顺序存储这些数据)
本题可以用动态规划的思路来解决,我们不难发现其实每次选取一个数为根节点存储的数据时,对应的形成的二叉树的数目等于各种可能情况下的左子树二叉树数量乘以右子树二叉树数量之和,比如n=3时,除去根节点还有两个节点,一共能形成二叉树的数目就是含有两个节点的左子树二叉树的可能数目乘以含有0个节点的右子树二叉树的可能数目加上含有一个节点的左子树二叉树的可能数目乘以含有一个节点的右子树二叉树的可能数目加上含有0个节点的左子树二叉树的可能数目乘以含有两个节点的右子树二叉树的可能数目。我发现本题与卡特兰数同样有着密切的关系。卡特兰数的递推式如下,Cn+1的值也恰好对应有n+1个数要存储时能有多少种二叉树:
代码如下:
class Solution {
public:
int numTrees(int n) {
if(n<=0) return 0;
int res[10001];
memset(res, 0, sizeof(res));
res[0] = 1;
res[1] = 1;
for(int i=2;i<=n;i++)
{
for(int j=0;j<i;j++)
{
res[i] += res[j]*res[i-j-1];
}
}
return res[n];
}
};
这里要注意当传入的n小于等于0时答案是0要直接返回0,后面定义res[0]等于0是因为当节点总数大于等于2时,遇到左子树或者右子树为空时就当有一种情况,如果res[0]当成0的话会对结果产生影响。
用动态规划的思路解决本题可以跳出解决二叉树问题时的一般思路,即不需要建立二叉树,也不需要对二叉树进行遍历。只需要进行纯数字的推倒,显然更简单一些。