题目:
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,所以我们能确定如果根为k,则根左边的数是1到k-1,根右边的数是k+1到n。还有一点技巧是,对于通过一个根来说,唯一二叉树的数量是其左子树的数量乘以右子树的数量,这是简单的乘法原理。并且,左右子树的形态数量是跟具体的数无关的,只跟这个树里有多少节点有关。而根可以选择从1到n的任意的数,唯一二叉树的总数,就是根为1到n的树相加。所以该问题化简为以k为根,其唯一左子树和右子树各有多少,这就是个动态规划的问题了。(这个解析来自LeetCode上大牛的解题思路,我这里翻译下)
所以这里又绕回来动态规划了,求动态规划问题,无非就是确定边界和转移公式
这里边界无非就是dp[0] =1,dp[1] =1这里很好理解
dp[0] 表示,当n=0时,此时为空树,这也是BST树
dp[1] 表示 当n=1时,此时只有一个根节点,那么数量也是为1
转移公式:dp[i] += dp[i-j-1]*dp[j]; 以这个结点为根的可行二叉树数量就是左右子树可行二叉树数量的乘积,所以总的数量是将以所有结点为根的可行结果累加起来
dp[2] = dp[1] * dp[0] + dp[0]*dp[1]
dp[3] = dp[2] *dp[1] + dp[1]*dp[1]+dp[0]*dp[2]
代码Java:
public static int numTrees(int n) {
if(n <= 0) {
return 0;
}
int[] dp = new int[n+1];
dp[0] = 1;
dp[1] = 1;
for(int i = 2;i <= n;i++) {
for(int j = 0; j < i;j++) {
dp[i] += dp[i-j-1]*dp[j];
}
}
return dp[n];
}