题目描述
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拿到这个题目,我是知道用动态规划的,但是找不到递推关系,查了很多博客以后,勉强得出一个答案,但是说实话说服力一般。
public class Solution {
public int numTrees(int n) {
if(n == 0)
return 1;
if(n == 1)
return 1;
int[] res = new int[n+1];
res[0] = 1;
res[1] = 1;
for(int i=2; i<=n; i++){
for(int k=0; k<i; k++){
res[i] += res[k] * res[i-1-k];
}
}
return res[n];
}
}
思路就是:反正就是BST,每个元素都做根节点,算出每个元素做根节点时有几种情况,然后每个节点的几种情况相加。这就是上述代码的原理。
每个元素有几种情况,根据bst的特性,就是中序便利是排序的。所以:
以n=3为例。新建一个int res[]= int[3+1](循环1,2,3,每个数字对应所再index,而不是1对应0index,2对应1 index...)
1是根节点的时候,1的左子树是0个,右子树是2,3(两个)。所以是1是根节点的种类数是:res[0]乘以res[2]情况数。
2是根节点的时候,2的左子树是1(1个),右子树是1(1个)。所以是2是根节点的种类数是:res[1]乘以res[1]情况数。
3是根节点的时候,3的左子树是1,2(2个),右子树是(0个)。所以是2是根节点的种类数是:res[2]乘以res[0]情况数。