LeetCode算法练习之不同的二叉搜索树II

分析题目，假设我们输入的是5，那么有一部分答案就是这样的：

这时很明显原问题的解取决于部分问题的解。故马上想到使用递归来求解。

继续分析问题，我们先选择一个根，比如2，然后以2为节点生成二叉搜索树。2的左子树可以是小于2的数生成的任意二叉搜索树，右子树是大于2的数生成的任意二叉搜索树。有了这个思路，就可以编写出代码了。

/**
 * 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){
            vector<TreeNode*> res;
            return res;
        }
        return gen(1,n);
    }
    vector<TreeNode*> gen(int start,int end){
        vector<TreeNode*> res;
        if(start > end) res.push_back(nullptr);
        for(int i=start;i<=end;i++){
            vector<TreeNode*> left = gen(start,i-1);
            vector<TreeNode*> right = gen(i+1,end);
            for(int l=0;l<left.size();l++){
                for(int r=0;r<right.size();r++){
                    TreeNode* root = new TreeNode(i);
                    root -> left = left[l];
                    root -> right = right[r];
                    res.push_back(root);
                }
            }
        }
        return res;
    }
};

 

这个题这样的分数可以了，但是还能再进行优化。凡是递归问题，都可以往动态规划那边想想，这个题也是这样。做完题目之后本来打算写博客完事，但是突然就想到了一个问题。

还是假设输入是5，考虑这两个二叉搜索树：1,2[3,4,5]和2,1[3,4,5]：

画出图来（懒得拍了），我们发现，不管是先插入1，还是先插入2，插入345的时候都要执行一遍递归运算。这样345这个子问题就算了两遍，造成了时间浪费！

批评LeetCode，这样一个不是最优的算法居然还拿到了一个高分，不应该！害我没有发现最优解。

解决也方便，把答案储存在一个二维数组中就可以了。但是这样造成的内存占用太大导致没有过测试。。。