题目
给你一个整数 n ,请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。
示例 1:
输入: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 <= 8
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<TreeNode> generateTrees(int n) {
List<TreeNode> list=new LinkedList<TreeNode>();
if(n==0)
return null;
if(n==1)
{
list.add(new TreeNode(1));
return list;
}
return dfsHelper(1, n);
}
private List<TreeNode> dfsHelper(int l, int r) {
List<TreeNode> list=new LinkedList<TreeNode>();
if(l>r)
{
list.add(null);
return list;
}
if(l==r)
{
list.add(new TreeNode(l));
return list;
}
for (int i = l; i <= r; i++) {
List<TreeNode> left = this.dfsHelper(l, i - 1);
System.out.println(" left"+left);
List<TreeNode> right = this.dfsHelper(i + 1, r);
System.out.println(" right "+right);
for (TreeNode val_l : left) {
for (TreeNode val_r : right) {
TreeNode node=new TreeNode(i);
node.left=val_l;
node.right=val_r;
list.add(node);
}
}
}
return list;
}
}
反思
1、和上一个类似,分治法
2、定义递归函数(左右边界)——结束条件(简单情况)——for循环,以每个i为根节点——对于每个根节点,递归调用左右子树,并遍历返回结果——加入结果