leetcode刷题笔记九十五题 不同的二叉搜索树II
leetcode刷题笔记九十五题 不同的二叉搜索树II
源地址:95. 不同的二叉搜索树 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提示:
0 <= n <= 8
/**
需要注意的是二叉搜索树性质,左侧小于根节点,右侧大于根节点
本题可以使用递归法或动态规划法,这里动态规划使用复杂offSet
主要是根据root位置, 调用左右子树位置的递归函数或DP数组结果
*/
/**
* Definition for a binary tree node.
* class TreeNode(_value: Int = 0, _left: TreeNode = null, _right: TreeNode = null) {
* var value: Int = _value
* var left: TreeNode = _left
* var right: TreeNode = _right
* }
*/
import scala.collection.mutable
object Solution {
def generateTrees(n: Int): List[TreeNode] = {
if(n == 0) return List()
return helper(1, n)
def helper(start: Int, end: Int): List[TreeNode]= {
var res = mutable.ListBuffer[TreeNode]()
//边界 补null
if(start > end){
res += null
return res.toList
}
//只有一个结点, 插入
if(start == end){
val root = new TreeNode(start)
res += root
return res.toList
}
//对左右子树调用递归函数,与root连接
for(i <- start to end){
val leftTree = helper(start, i-1)
val rightTree = helper(i+1, end)
for(leftElem <- leftTree){
for(rightElem <- rightTree){
val root = new TreeNode(i)
root.left = leftElem
root.right = rightElem
res += root
}
}
}
return res.toList
}
if(n == 0) return List()
return helper(1, n)
}
/**
动态规划与递归思想基本一致
初始状态 dp(0) = null
动态转换方程 1 <= root <= n时 dp(root) = root + dp(root-1) + dp(root+1)
需要注意的时dp(root+1)之前并没有得到计算,但是由树的构成特点可知,树的形状只与结点个数有关,相同长度的树的构成是一致,只需要对值进行修改即可
故动态转换方程转变为 1 <= root <= n时 dp(root) = root + dp(root-1) + clone(dp(length-root), root)
其中root-1为左子树节点数,length - root为右子树结点数, 此时根节点为root,由于右侧结点均大于root, 需对其全部加root
*/
/**
* Definition for a binary tree node.
* class TreeNode(_value: Int = 0, _left: TreeNode = null, _right: TreeNode = null) {
* var value: Int = _value
* var left: TreeNode = _left
* var right: TreeNode = _right
* }
*/
import scala.collection.mutable
object Solution {
def generateTrees(n: Int): List[TreeNode] = {
def clone(n: TreeNode, offSet: Int): TreeNode = {
if (n == null) return null
val node = new TreeNode(n.value + offSet)
node.left = clone(n.left, offSet)
node.right = clone(n.right, offSet)
return node
}
//注意初始化问题
if(n == 0) return List()
val dp: Array[mutable.ListBuffer[TreeNode]] = Array.fill(n+1)(mutable.ListBuffer())
dp(0) = mutable.ListBuffer(null)
for(length <- 1 to n ){
for(rootPos <- 1 to length){
val left = rootPos - 1
val right = length - rootPos
for (leftElem <- dp(left); rightElem <- dp(right)){
val root = new TreeNode(rootPos)
root.left = leftElem
root.right = clone(rightElem, rootPos)
dp(length) += root
}
}
}
return dp(n).toList
}
}