题目
给定一个整数 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 种不同结构的二叉搜索树:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees-ii。
代码及分析
二叉搜索树,又被称为是二叉查找树、二叉排序树,这种树其左子树所有节点的值均小于根节点的值,其右子树所有节点的值均大于根节点的值,并且其左、右子树均为二叉搜索树。
空树也是二叉搜索树。
方法一:递归法
与这道题类似的是那道“96不同的二叉搜索树”(https://blog.csdn.net/weixin_44412429/article/details/95481895)
在96中我们曾分析到,由于给定的节点值为1~n,是一组从小到大顺序排列的数据,其中,任意一个值均可以作为根节点出现,记为 i,而 i 的取值范围为1 ~ n。
我们可以采用递归的想法,遍历所有节点值,将其作为根节点,当以 i 为根节点时,为了满足搜索树的性质,其左子树必定有1 ~ i-1 构成,右子树必定由 i+1 ~ n 构成。然后对于左子树的所有节点数1~ i-1递归地也构造成二叉搜索树,同理,右子树类似。
vector<TreeNode*> generateTrees(int n) {
vector<TreeNode*> res;
if(n == 0)
return res;
return fun(1,n);
}
vector<TreeNode*> fun(int start, int end)
{
vector<TreeNode*> vec;
if(start > end)
return {NULL};
for(int i = start; i <= end; i++)
{
vector<TreeNode*> left = fun(start,i-1);
vector<TreeNode*> right = fun(i+1,end);
int l1 = left.size();
int l2 = right.size();
for(int j = 0; j < l1; j++)
for(int k = 0; k < l2; k++)
{
TreeNode* root = new TreeNode(i);
root->left = left[j];
root->right = right[k];
vec.push_back(root);
}
}
return vec;
}