1、题目
给你一个整数 n ,请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。
链接: https://leetcode-cn.com/problems/unique-binary-search-trees-ii/
2、思路分析
1)回溯
我们定义 generateTrees(start, end) 函数表示当前值的集合为 [start,end],返回序列 [start,end] 生成的所有可行的二叉搜索树。
我们考虑枚举 [start,end] 中的值 i 为当前二叉搜索树的根,那么序列划分为[start,i−1] 和[i+1,end] 两部分。我们递归调用这两部分,即 generateTrees(start, i - 1) 和 generateTrees(i + 1, end),获得所有可行的左子树和可行的右子树。
那么最后一步我们只要从可行左子树集合中选一棵,再从可行右子树集合中选一棵拼接到根节点上,并将生成的二叉搜索树放入答案数组即可。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<TreeNode*> generateTrees(int start, int end)
{
if(start > end){
return {nullptr};
}
if(start == end){
return {new TreeNode(start)};
}
vector<TreeNode*> res;
for(int i = start; i <= end; ++i){
// 左子树集合
vector<TreeNode*> left = generateTrees(start, i - 1);
vector<TreeNode*> right = generateTrees(i + 1, end);
for(auto & l :left){
for(auto& r : right){
res.push_back(new TreeNode(i, l, r));
}
}
}
return res;
}
vector<TreeNode*> generateTrees(int n) {
return generateTrees(1, n);
}
};