给定一个整数 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
使用递归的思想:
我们要怎么确定递归的函数呢?仔细观察上面的样例,我们可以发现,其实我们可以理解成循环遍历每一个数(1,2,3),将每次遍历到的数设为根节点root,然后根据BST(二叉搜索树)的特性进行递归:比root小的值构成左子树,比root大的值构成右子树,然后再对生成的子树看做一条独立的数进行递归,那么,递归终止条件是什么呢?递归终止条件就是当只剩下一个节点的时候(叶子),结束递归,下面是java递归解决问题的代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<TreeNode> generateTrees(int n) {
if(n <= 0){
return new ArrayList<>();
}
return generateTrees(1, n);
}
public List<TreeNode> generateTrees(int start, int end) {
if(start == end){
List<TreeNode> list = new ArrayList<>();
TreeNode node = new TreeNode(start);
list.add(node);
return list;
}
//已mid为根节点 左右子树分别为start到mid-1 和 mid+1到end 注意边界
List<TreeNode> res = new ArrayList<>();
for(int mid = start; mid <= end; mid++){
List<TreeNode> left = new ArrayList<>();
List<TreeNode> right = new ArrayList<>();
if(start != mid){
//说明有左子树
left = generateTrees(start, mid - 1);
}
if(end != mid){
//说明有右子树
right = generateTrees(mid + 1, end);
}
if(!left.isEmpty() && !right.isEmpty()){
//生成BST
for(TreeNode leftTree : left){
for(TreeNode rightTree : right){
TreeNode root = new TreeNode(mid);
root.left = leftTree;
root.right = rightTree;
res.add(root);
}
}
}else if(left.isEmpty()){
for(TreeNode rightTree : right){
TreeNode root = new TreeNode(mid);
root.right = rightTree;
res.add(root);
}
}else if(right.isEmpty()){
for(TreeNode leftTree : left){
TreeNode root = new TreeNode(mid);
root.left = leftTree;
res.add(root);
}
}
}
return res;
}
}