题目要求
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
解题思路
BST指的是二分搜索树,比跟节点小的在左,比跟节点大的在右。
当节点为n时,不同的BST包括当节点分别为1,2,3,...,n时的BST之和。
对于n个节点,当跟节点为i时,比i小的i-1个节点作为左子树,比i大的(n-1)-(i-1)=n-i个节点作为右子树。左子树的方案数*右子树的方案数为总的方案数。
定义:F(i, n) = G(i-1) * G(n - i),
F(i, n)的含义是:以i为根节点,且总共有n个节点的树。例如F(2,3)就是说2为根节点,总共有3个节点。
递推的初始条件为G(0)=1, G(1)=1。
复杂度分析
时间复杂度为O(N*N)
代码如下
class Solution {
public:
int numTrees(int n) {
vector<int>dp(n+1,0);
dp[0]=1;
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];
}
};