[路飞]_leetcode 95. 不同的二叉搜索树 II

74 阅读1分钟

题目来源:leetcode 95. 不同的二叉搜索树 II

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

 

示例 1:

输入: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 <= 8

代码实现

/**
 * 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[]}
 */
function build(lo, hi) {
  let res = [];
  if (lo > hi) {
    res.push(null);
    return res;
  }
  for (let i = lo; i <= hi; i++) {
    let leftTree = build(lo, i - 1);
    let rightTree = build(i + 1, hi);
    for (let left of leftTree) {
      for (let right of rightTree) {
        let root = new TreeNode(i);
        root.left = left;
        root.right = right;
        res.push(root);
      }
    }
  }
  return res;
}
var generateTrees = function (n) {
  if (n == 0) return [];
  return build(1, n);
};