【leetcode-树】不同的二叉搜索树 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,其步骤如下:

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

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 generate(1, n);
    }
    
    
    private List<TreeNode> generate(int start, int end) {
        List<TreeNode> res = new ArrayList<>();
        if(start > end) {
            res.add(null);
            return res;
        }
        
        for(int i = start; i <= end; i ++) {
            List<TreeNode> lefts = generate(start, i - 1);
            List<TreeNode> rights = generate(i + 1, end);
            for(TreeNode left : lefts) {
                for(TreeNode right : rights) {
                    TreeNode node = new TreeNode(i);
                    node.left = left;
                    node.right = right;
                    res.add(node);
                }
            }
        }
        
        return res;
    }
}

由于水平有限,文章中难免会有一些错误,有纰漏之处恳请各位大佬不吝赐教!

及时更新最新文章和学习资料,一起来学习:

推荐阅读:

【leetcode-数组】买卖股票的最佳时机 II  - CSDN博客

 【leetcode-数组】存在重复

 【leetcode-数组】两数之和 II - 输入有序数组

【leetcode-数组】长度最小的子数组 

【leetcode-数组】 旋转数组 

猜你喜欢

转载自blog.csdn.net/kangbin825/article/details/106018130
今日推荐