题目
给定一个整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
1.分治法:每个i结点的时候 当做root,然后分别递归计算其左右子树的二叉搜索树。
2.思路有了,怎么输出也是个问题:
代码
/**
* 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 LinkedList<TreeNode>();
}
return generateTreess(1,n);
}
public List<TreeNode> generateTreess(int start,int end){
List<TreeNode> res = new LinkedList<>();
//出口
if(start>end){
res.add(null);
return res;
}
for(int i=start;i<=end;i++){
List<TreeNode> left = generateTreess(start,i-1);
List<TreeNode> right = generateTreess(i+1,end);
for(TreeNode x:left){
for(TreeNode y:right){
TreeNode node = new TreeNode(i);
node.left=x;
node.right=y;
res.add(node);
}
}
}
return res;
}
}