【leetcode】95 不同的二叉搜索树II(二叉树)

探讨LeetCode上一道经典题目,如何生成由1到n构成的所有可能的二叉搜索树。通过递归方法,枚举每个可能的根节点,生成左子树和右子树的所有组合。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接:https://leetcode-cn.com/problems/unique-binary-search-trees-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到n枚举每一个值作为根节点,把每次生成的二叉树结构的根节点都保存下来

假设其中一次是以i为根节点,树的节点值为begin <= i <= end,其步骤如下:

  1. 用{begin, i-1}递归生成左子树的所有结构,并将所有结构的根节点保存在lSubs
  2. 用{i+1,end}递归生成右子树的所有结构,并将所有结构的根节点保存在rSubs
  3. 在以i为根节点的前提下,遍历lSubs和rSubs的每一个节点,将i节点值生成的新节点作为他们的根节点;将该结构存入返回列表

代码

class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        if(n <=0) 
            return vector<TreeNode*> {};
        return generate(1, n);
    }
    
    vector<TreeNode*> generate(int begin, int end){
        vector<TreeNode*> ret;
        if(begin > end){
            ret.push_back(nullptr); // 注意:必须加入空指针
            return ret;
        }
            
        for(int i = begin; i<=end; ++i){
            
            vector<TreeNode*> lSubs = generate(begin,i-1);  // {begin,i-1}所有结构的根节点列表
            vector<TreeNode*> rSubs = generate(i+1, end);   // {i+1, end}所有结构的根节点列表
            for(auto l: lSubs){
                for(auto r: rSubs){
                    TreeNode *root = new TreeNode(i);
                    root->left = l;
                    root->right = r;
                    ret.push_back(cloneTree(root));
                }
            }
        }
        
        return ret;
    }
};

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值