LeetCode解题报告 96. Unique Binary Search Trees [medium]

题目要求

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];
    }
};