leetcode 95. Unique Binary Search Trees

题目

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

分析：

这个题的思想和《算法导论》上的“矩阵链乘法”和“钢条割据问题”非常类似！

代码一：100.00%  5.38%

自底而上的动态规划。

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

题目 95. Unique Binary Search Trees II

Given an integer n, generate all structurally unique BST's (binary search trees) that store values 1 ... n.

分析：相对上个题目而言，需要生成问题的解结构。

代码一：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        if(n==0) return {};
       vector<vector<vector<TreeNode*>>> dp(n+2,vector<vector<TreeNode*>>
                                        (n+2,vector<TreeNode*>(1,NULL)));
        for(int j=1;j<=n;++j){
            for(int i=j;i>=1;--i){
                dp[i][j].clear();
                for(int k=i;k<=j;++k){
                    for(int k1=0;k1<dp[i][k-1].size();++k1){
                        for(int k2=0;k2<dp[k+1][j].size();++k2){
                            TreeNode *temp=new TreeNode(k);
                            temp->left=dp[i][k-1][k1];
                            temp->right=dp[k+1][j][k2];
                            dp[i][j].push_back(temp);
                        }
                    }
                }
            }
        }
        return dp[1][n];
        }
    
};