动态规划——95. 不同的二叉搜索树 II

难度：中等

题目

给定一个整数 n，生成所有由 1 … n 为节点所组成的二叉搜索树。

示例:

输入: 3
输出:
[
[1,null,3,2],
[3,2,null,1],
[3,1,null,null,2],
[2,1,3],
[1,null,2,null,3]
]
解释:
以上的输出对应以下 5 种不同结构的二叉搜索树：

解答

思路：选择根节点，把小于根节点的放到左边，把大于根节点的放到右边。递归，先构建底部的节点，再搭建起来。

时间复杂度：O(n2)

空间复杂度：O(n2)

代码

class Solution {
public:
    vector<TreeNode*> generateTreesCore(int start,int end){
        vector<TreeNode*> temp;
        if(start>end){
            temp.push_back(nullptr);
        }
        for(int i=start;i<=end;i++){
            vector<TreeNode*> left=generateTreesCore(start,i-1);
            vector<TreeNode*> right=generateTreesCore(i+1,end);
            for(auto l:left){
                for(auto r:right){
                    TreeNode* node=new TreeNode(i);
                    node->left=l;
                    node->right=r;
                    temp.push_back(node);
                }
            }
        }
        return temp;
    }

    vector<TreeNode*> generateTrees(int n){
        return n==0?(vector<TreeNode*>)0 : generateTreesCore(1,n);
    }
};