不同的二叉搜索树

不同的二叉搜索树

描述

给定一个整数 n，求以 1 … n 为节点组成的二叉搜索树有多少种？

示例:

输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:

1 3 3 2 1
\ / / / \
3 2 1 1 3 2
/ / \
2 1 2 3

思路和代码

动态规划的思路，假设有0个数，那么只有一种可能，假设1个数，那么也只有一种可能。
对于其他任意的 i ，说明有1到i个数，假设以1作为根节点，则左子树有0个，右子树有 i-1个，那么就有dp[0]dp[i-1]种可能。如果以2作为根节点，则左子树就是一个结点，右子树就是 i-2个结点，以此类推dp[ i ] += dp[ j ] dp[ i-j ]，其中j从1到 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 j=1; j<=i; j++)
                dp[i] += dp[j-1] * dp[i-j];
        }
        return dp[n];
    }
};

不同的二叉搜索树2

不同于上一题，上一题只要求有多少个不同的二叉搜索树，这一题还需要将不同的二叉搜索树保存下来

思路和代码

还是递归的思路，对于根节点为k的情况，则递归得到左子树的所有可能组合，再递归得到右子树的所有可能情况，然后再组合左右子树的所有可能，然后挂在根节点下面

class Solution {
public:
    vector<TreeNode *> generateTrees(int n) { 
        vector<TreeNode*> res;
        if (n == 0) 
            return res; 
        return generate(1, n); 
    } 
private: 
    vector<TreeNode *> generate(int start, int end) { 
        vector<TreeNode*> subTree; 
        if (start > end) { 
            subTree.push_back(nullptr); 
            return subTree; 
        } 
        for (int k = start; k <= end; k++) { 
            vector<TreeNode*> leftSubs = generate(start, k - 1); 
            vector<TreeNode*> rightSubs = generate(k + 1, end); 
            for (auto i : leftSubs) { 
                for (auto j : rightSubs) { 
                    TreeNode *node = new TreeNode(k); 
                    node->left = i; 
                    node->right = j; 
                    subTree.push_back(node); 
                } 
            } 
        }
        return subTree; 
    } 
};