【LeetCode】96. Unique Binary Search Trees

96. Unique Binary Search Trees

Description：
Given n, how many structurally unique BST’s (binary search trees) that store values 1 … n?
Difficulty：Medium

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

方法1:动态规划

• Time complexity : $O\left(n^2\right)$
• Space complexity : $O\left(n\right)$
  思路：
  枚举[1,2......n]中每个元素作为根节点，左右两边可以递归进行求解，最后累加起来的总和即为答案，但是为了避免不必要的重复计算，使用动态规划进行求解。
  首先有一个隐藏条件，长度等长的数组能构造出来的BTS数量一样，例如[1,2,3]和[4,5,6]
G(n)表示长度为n的数组可以组成BTS的数量
  F(i, n)表示i作为根节点的BTS数量
  $G(n)=\sum F(i,n)$
  $F(i,n)=G(i-1)\cdot G(n-i)$
  $G(n)=\sum (i-1)\cdot G(n-i)$

class Solution {
public:
    int numTrees(int n) {
        vector<int> res(n+1, 0);
        res[0] = 1;
        res[1] = 1;
        for(int i = 2; i <= n; i++)
            for(int j = 1; j <= i; j++)
                res[i] += res[j-1] * res[i-j];
        return res[n];
    }
};

方法2：数学推导

• Time complexity : $O\left(n\right)$
• Space complexity : $O\left(1\right)$
  思路：
  $C_0=1,C_{n+1}=\frac{2(2n+1)}{n+2}{C}_n$

class Solution(object):
    def numTrees(self, n):
        C = 1
        for i in range(0, n):
            C = C * 2*(2*i+1)/(i+2)
        return int(C)