题目链接:https://leetcode-cn.com/problems/unique-binary-search-trees/
解题步骤:
题目可以转换为求以i(1 <= i <= n)为根节点的不同结构的二叉树数量,然后相加就可以的出答案。假如我们以i为根节点,左子树由[ 1, i-1]构成,右子树由[i+1, n]构成。由于二叉搜索树左节点值小于根节点值,右节点值大于根节点值,并对所有子树都成立,所以左子树[1, i-1]与右子树[i+1, n]与原问题[1, n]是一样的,因此我们使用动态规划来求解:
- 定义dp[n]为[1,n]产生不同二叉搜索树的数量
- 初始化基本情况, dp[0] = dp[1] = 1
- 转换公式:dp[i] = dp[j-1] * dp[i-j] (1 <= j <= i)
C++实现:
class Solution {
public:
int numTrees(int n) {
vector<int> dp(n+1);
dp[0] = dp[1] = 1;
for (int i = 2; i <= n; i++) {
for (int j = 1; j<= i; j++) {
dp[i] += dp[j-1] * dp[i-j];
}
}
return dp[n];
}
};