96. Unique Binary Search Trees

问题描述:

Given n, how many structurally unique BST's (binary search trees) that store values 1 ... n?

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

解题思路:

这道题可以用动态规划来解:

若有n个数字,每一个数字都可能是根节点,从中抽取   1 ≤ i ≤ n

可以抽象出一棵树:

               

左子树为由 [1, i-1]构成的树,有i-1个节点

右子树为由[i+1, n]构成的树,有n-i个节点

所以状态转移方程为:

dp[i] = Σ(dp[k] * dp[i-k-1])  1 ≤ k ≤ i

代码:

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 k = 0; k < i; k++){
                dp[i] += dp[k] * dp[i - k - 1];
            }
        }
        return dp[n];
    }
};
原文地址:https://www.cnblogs.com/yaoyudadudu/p/9158354.html