原题
思路
- 动态规划
- 有n个节点的不同二叉搜索树的数量其实就是依次从值最小的节点开始作为根节点,那么左右两边的节点数分别为0,n-1。那么此时的以值最小的节点作为根节点的不同二叉树的数量为dp[0]*dp[n-1]。依次遍历为dp[1]*dp[n-2]…把所有的数量加起来,总和即为dp[n]。
代码
class Solution {
public:
int numTrees(int n) {
if (n <= 2) return n;
int dp[n+1];
dp[0] = 1;
dp[1] = 1;
dp[2] = 2;
for (int i = 3; i <= n; i++) {
dp[i] = 0;
for (int j = 0; j <= i-1; j++) dp[i] += dp[j] * dp[i-j-1];
}
return dp[n];
}
};
运行截图
收获
- 动态规划不仅是考虑n和n-1之间的问题,更是考虑它和之前已经存在的数值的关系。