题意:
给你一个整数 n,请你生成并返回所有由n个节点组成且节点值从1到n
互不相同的不同 二叉搜索树
可以按 任意顺序 返回答案。
示例
实现
这里简化理论的支持,直接上代码。
我们选择一个数作为根节点,然后分配左右子树,最后按照左右子树的大小继续递归建树即可得到答案。
import java.util.LinkedList;
import java.util.List;
/**
* @ClAssName DifferentTree
* @Description 不同的树
* @Author 欧妮甲是神仙
* @Date 2024/9/2 22:{MINUTE}
*/
public class DifferentTree {
//创建节点
class TreeNode{
int val ;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
}
public TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
public List<TreeNode> generateTrees(int n) {
//主方法,接受一个整数n作为参数,表示生成树的数量
if (n==0){
// n为0时,没有可生成的树,返回一个空列表
return new LinkedList<>();
}
return generateRandomBST(1, n);
}
//辅助方法,接收两个整数作为参数,表示生成树值的范围
public List<TreeNode> generateRandomBST(int start, int end) {
//用于存储所有生成树的列表
LinkedList<TreeNode> trees = new LinkedList<>();
// 当开始值大于结束值时,表示该区间无有效数字,返回一个包含nul1的列表
if ( end <start) {
trees.add(null);
return trees;
}
//遍历从start到end的所有整数,将其每一个整数作为根节点
for (int i = start; i <= end; i++) {
//递归生成左子树和右子树
List<TreeNode> leftTrees = generateRandomBST(start, i - 1);
List<TreeNode> rightTrees = generateRandomBST(i + 1, end);
//遍历左子树和右子树的所有组合
//建树
for (TreeNode leftTree : leftTrees) {
for (TreeNode rightTree : rightTrees) {
TreeNode treeNode = new TreeNode(i);
treeNode.left = leftTree;
treeNode.right = rightTree;
trees.add(treeNode);
}
}
}
return trees;
}
}
演示:
当输入n=3时,我们需要构造所有可能的二又搜索树,其中树的节点值从1到3。
遍历每个数字 i从1到 3,将其视为根节点,并递归地为其左右子树生成所有可能的结构。
1.i=1作为根节点
- 左子树:无
- 右子树:2,3
- 当2作为根节点:左子树无,右子树有 3
- 当3作为根节点:左子树有 2,右子树无
结果:
2. i=2 作为根节点:
- 左子树:1
- 右子树:3
结果:
3. i=3 作为根节点
- 左子树:1,2
- 当1作为根节点:右子树有 2,左子树无·
- 当2作为根节点:左子树有 1,右子树无
- 右子树: 无
结果:
以上能够看到有5种不同搜索树。
总结
这题的核心思想是采用递归的方法来生成所有可能的二又搜索树。
1.递归基础:
当给定的范围内没有数字时(即 start>end)这意味着此处的子树为nul,所以我们应该返回一个列表,其中只包含一个null元素。
2中间节点选择:
对于范围内的每个数字 i,我们都尝试将其作为当前树的根节点。也就是说,我们认为每个数字都有可能作为根节点。
3.左、右子树的生成:
- 于根节点 i,它的左子树应该由 start 到 i-1的所有可能树组成,而右子树应该由 i +1到 end 的所有可能树组成。
- 使用递归,我们可以得到两个列表:leftTrees和rightTrees ,分别代表所有可能的左和右子树。
- 然后,我们组合这些左子树和右子树,形成所有可能的树结构。