问题:
给你一个整数 n ,请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。
示例:
输入:n = 3
输出:[[1,null,2,null,3],[1,null,3,2],[2,1,3],[3,1,null,null,2],[3,2,null,1]]
示例 2:
输入:n = 1
输出:[[1]]
思想:
回溯法:二叉搜索树的结构特点是左子树小于根节点,右子树大于根节点,所以我们要把1到n的数分成左右两部分,然后递归的找出所有左子树和右子树。然后随机的组合左右子树。
代码:
public List<TreeNode> generateTrees(int n) {
if(n ==0) return new ArrayList<TreeNode>();
return getsubtree(1,n);
}
public List<TreeNode> getsubtree(int start,int end){
List<TreeNode> lhl = new ArrayList<TreeNode>();
if(start>end){
lhl.add(null);
return lhl;
}
for(int i = start;i<=end;i++){
List<TreeNode>leftnode = getsubtree(start,i-1);
List<TreeNode>rightnode = getsubtree(i+1,end);
for(TreeNode left:leftnode){
for(TreeNode right:rightnode){
TreeNode root = new TreeNode(i);
root.left = left;
root.right = right;
lhl.add(root);
}
}
}
return lhl;
}