[LeetCode]95. 不同的二叉搜索树 II(java实现)dfs
1. 题目
2. 读题(需要重点注意的东西)
思路(dfs):
二叉搜索树: 左孩子比根节点小,右孩子比根节点大,即中序遍历是有序的
由1~n的数能生成的二叉搜索树的数量是卡特兰数:
由于二叉搜索树的中序遍历是有序的,那么二叉搜索树中的每一颗子树都是连续的一段:
- 对于每段连续的序列,枚举二叉搜索树根节点的位置
- 分别递归求出左右子树的方案(即合法的左右子树根节点的位置)
- 左子树的任意一种方案和右子树的任意一种方案拼在一起,可以得到当前节点的一种方案,所以我们将左右子树的所有方案两两组合,并记录在答案中
3. 解法
---------------------------------------------------解法---------------------------------------------------:
/**
* 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) {
if(n == 0) return new ArrayList<TreeNode>();
return dfs(1,n);
}
private List<TreeNode> dfs(int l,int r){
// 注意,list中存储的是根节点,最后输出的是以该根节点开始层次遍历的树
List<TreeNode> list = new ArrayList<>();
if(l > r){
list.add(null);
return list;
}
// 对于每一个以i作为的根节点
for(int i = l;i <= r;i++){
// 递归出其左子树的所有情况
List<TreeNode> left = dfs(l,i-1);
// 递归出其右子树的所有情况
List<TreeNode> right = dfs(i+1,r);
// 左右任选一组合即为一个合法情况
for(TreeNode a : left)
for(TreeNode b : right){
// 对于每一种情况,都要new一个新的节点来存储它
TreeNode root = new TreeNode(i);
root.left = a;
root.right = b;
list.add(root);
}
}
return list;
}
}
可能存在的问题:
4. 可能有帮助的前置习题
5. 所用到的数据结构与算法思想
- dfs