分析题目,假设我们输入的是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,这样一个不是最优的算法居然还拿到了一个高分,不应该!害我没有发现最优解。
解决也方便,把答案储存在一个二维数组中就可以了。但是这样造成的内存占用太大导致没有过测试。。。