给定一个整数 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
代码
public List<TreeNode> generateTrees(int n) {
if (n == 0) return new ArrayList<TreeNode>();
return generate(1, n);
}
public List<TreeNode> generate(int left, int right) {
if (left > right) return new ArrayList<TreeNode>() {
{
add(null);
}
};
if (left == right) return new ArrayList<TreeNode>() {
{
add(new TreeNode(left));
}
};
ArrayList<TreeNode> res = new ArrayList<>();
for (int i = left; i <= right; i++) {
List<TreeNode> leftTree = generate(left, i - 1);
List<TreeNode> rightTree = generate(i + 1, right);
for (TreeNode l : leftTree)
for (TreeNode r : rightTree) {
TreeNode temp = new TreeNode(i);
temp.left = l;
temp.right = r;
res.add(temp);
}
}
return res;
}