Given n, how many structurally unique BST's (binary search trees) that store values 1 ... n?
Example:
Input: 3
Output: 5
Explanation:
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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
柳神的思路:
1. 令v[i]表示i个数能构成的二叉树的个数
2. 当i>=3时,若以j为root结点,v[j-1]等于root结点左边的j-1个结点能构成的BST个数,v[i-j]等于root结点右边i-j个结点能构成的BST个数
3. v[j-1] * v[i-j]等于以j为root结点能构成的BST种数
j可以取1~i中的任意一个值,把这些所有计算出来的总数相加就是v[i]的值
class Solution {
public:
int numTrees(int n) {
int v[n+1] = {0};
v[0] = 1;
for(int i = 1; i <= n; i++){
if(i <= 2) v[i] = i;
else
{
for(int j = 1; j <= i; j++){
v[i] += v[j-1] * v[i-j];
}
}
}
return v[n];
}
};