【leetcode-树】不同的二叉搜索树 II

【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 generateTrees(int n) {
        if(n == 0) return new ArrayList<>();
        
        return generate(1, n);
    }
    
    
    private List generate(int start, int end) {
        List res = new ArrayList<>();
        if(start > end) {
            res.add(null);
            return res;
        }
        
        for(int i = start; i <= end; i ++) {
            List lefts = generate(start, i - 1);
            List 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)