[路飞]_leetcode刷题_95. 不同的二叉搜索树 II

119 阅读1分钟

题目

#### 95. 不同的二叉搜索树 II

给你一个整数 n ,请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 **。可以按 任意顺序 返回答案。

示例 1:

image.png

输入:n = 3
输出:[[1,null,2,null,3],[1,null,3,2],[2,1,3],[3,1,null,null,2],[3,2,null,1]]

示例 2:

输入:n = 1
输出:[[1]]

解法

思路

递归。

我们要求[1,n]个节点可以构造的树,可以细分的去看这个问题

以1为根节点,[]为左子树,[2,n]为右子树看有多少中构造可能 以2为根节点,[1]为左子树,[3,n]为右子树看有多少种可能 以3为根节点,[1,2]为左子树,[4,n]为右子树看有多少种可能 ....

然后再看第一种情况,看他的右子树[2,n]

看以2为根节点,[]为左子树,[3,n]为右子树看有多少种可能 .... 这就找到了递归的感觉了

递归的跳出条件为start > end

代码如下

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {number} n
 * @return {TreeNode[]}
 */
var generateTrees = function(n) {
    if (n == 0) {
        return [];
    }
    return helper(1, n);
};

function helper(start, end) {
        let allTrees = [];
        if (start > end) {
            allTrees.push(null);
            return allTrees;
        }

        // 枚举可行根节点
        for (let i = start; i <= end; i++) {
            // 获得所有可行的左子树集合
            let leftTrees = helper(start, i - 1);
            // 获得所有可行的右子树集合
            let rightTrees = helper(i + 1, end);
            // 从左子树集合中选出一棵左子树,从右子树集合中选出一棵右子树,拼接到根节点上
            for (let left of leftTrees) {
                for (let right of rightTrees) {
                    let currTree = new TreeNode(i);
                    currTree.left = left;
                    currTree.right = right;
                    allTrees.push(currTree);
                }
            }
        }
        return allTrees;
    }