LeetCode-95-不同的二叉搜索树II
LeetCode-95-不同的二叉搜索树II
题目
给定一个整数 n,生成所有由 1 ... n 为节点所组成的二叉搜索树。
给定一个整数 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 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
思路解析
由于之前做过这道题的上一题,96.不同的二叉搜索树,当时采用的非递归,所以我就顺其自然的考虑非递归解决问题。这里有一个特别麻烦的地方,在第96题的解释中,我描述到,当添加右子树的时候,会出现差值,当时只需要获取数量,所以可以不用管,但是这次需要输出搜索树,就不得不管了。
我的解决方案是,将右子树加上根节点的值,但是实际操作的时候遇到了浅层拷贝的问题,就是我需要复制出右子树的结构体,但是直接用=赋值的话,结构体中的指针变量所指向的值并不会拷贝,导致后续会出错;
但是要自己写一个深度拷贝感觉有点麻烦,背离了刷leetcode的本意,于是我瞄了一眼答案,
看到了大佬写的递归的代码,感叹我应该超越不了了,干脆偷懒到底直接提交。。。。
代码
class Solution {
public:
vector<TreeNode*> generateTrees(int n) {
if(n == 0) return {};
return helper(1, n);
}
vector<TreeNode*> helper(int start, int end) {
if(start > end) return {nullptr};
vector<TreeNode*> res;
for(int i=start; i<=end; i++) {
auto left = helper(start, i-1), right = helper(i+1, end);
for(auto a:left) {
for(auto b:right) {
TreeNode* node = new TreeNode(i);
node->left = a;
node->right = b;
res.push_back(node);
}
}
}
return res;
}
};