leetcode-95 不同的二叉搜索树 II

给定一个整数 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

使用递归的思想:

我们要怎么确定递归的函数呢?仔细观察上面的样例,我们可以发现,其实我们可以理解成循环遍历每一个数(1,2,3),将每次遍历到的数设为根节点root,然后根据BST(二叉搜索树)的特性进行递归:比root小的值构成左子树,比root大的值构成右子树,然后再对生成的子树看做一条独立的数进行递归,那么,递归终止条件是什么呢?递归终止条件就是当只剩下一个节点的时候(叶子),结束递归,下面是java递归解决问题的代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<TreeNode> generateTrees(int n) {
        if(n <= 0){
            return new ArrayList<>();
        }
        return generateTrees(1, n);
    }

    public List<TreeNode> generateTrees(int start, int end) {
        if(start == end){
            List<TreeNode> list = new ArrayList<>();
            TreeNode node = new TreeNode(start);
            list.add(node);
            return list;
        }
        
        //已mid为根节点 左右子树分别为start到mid-1 和 mid+1到end 注意边界
        List<TreeNode> res = new ArrayList<>();
        for(int mid = start; mid <= end; mid++){
            List<TreeNode> left = new ArrayList<>();
            List<TreeNode> right = new ArrayList<>();
            if(start != mid){
                //说明有左子树
                left = generateTrees(start, mid - 1);
            }
            if(end != mid){
                //说明有右子树
                right = generateTrees(mid + 1, end);
            }
            
            if(!left.isEmpty() && !right.isEmpty()){
                //生成BST
                for(TreeNode leftTree : left){
                    for(TreeNode rightTree : right){
                        TreeNode root = new TreeNode(mid);
                        root.left = leftTree;
                        root.right = rightTree;
                        res.add(root);
                    }
                }
            }else if(left.isEmpty()){
                 for(TreeNode rightTree : right){
                    TreeNode root = new TreeNode(mid);
                    root.right = rightTree;
                    res.add(root);
                 }
            }else if(right.isEmpty()){
              
                for(TreeNode leftTree : left){
                    TreeNode root = new TreeNode(mid);
                    root.left = leftTree;
                    res.add(root);
                }
            }
        }
        return res;
    }

}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值