题目链接:https://leetcode-cn.com/problems/unique-binary-search-trees/
题目大意:给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
思路:动态规划法。设v[ i ]表示1~i个节点组成的BST个数。从1~i中任选一个数 j 作为二叉搜索树的根节点,则以 j 为根节点的BST个数为,以1~j - 1为节点组成的BST个数v[j-1] 乘以 j+1 ~ i 为节点组成的BST个数。按照对称的思想,以 j+1 ~ i 为节点组成的BST个数就是从1~i - j 个节点构成的BST个数即v[i] += v[j - 1] * v[i - j]。最终的结果为v[n]。
代码:
class Solution {
public:
int numTrees(int n) {
vector<int> v(n + 1);
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];
}
};