0x00 题目
给你一棵所有节点为非负值的【二叉搜索树】,请你计算树中任意两节点的【差】的绝对值的【最小值】。
0x01 思路
二叉搜索树【中序遍历】是一个【升序】数组,可知,最小值只可能是相邻节点之差。
0x02 解法
语言:Swift
树节点:TreeNode
public class TreeNode {
public var val: Int
public var left: TreeNode?
public var right: TreeNode?
public init() { self.val = 0; self.left = nil; self.right = nil; }
public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
self.val = val
self.left = left
self.right = right
}
}
解法一:
中序遍历得到一个升序数组
遍历数组计算相邻两数之差
同时存储最小值
时间复杂度:O(N)
空间复杂度:O(N)
func getMinimumDifference(_ root: TreeNode?) -> Int {
var arr = [Int]()
// 中序遍历方法
func add(_ root: TreeNode?) {
guard let r = root else { return }
add(r.left)
arr.append(r.val)
add(r.right)
}
// 获取升序数组
add(root)
if arr.count < 1 {
return 0
}
// 计算差值
var ans = Int.max
var temp = 0
for i in 0..<arr.count - 1 {
temp = arr[i+1] - arr[i]
if temp < ans {
ans = temp
}
}
return ans
}
解法二:
中序遍历的同时,记录上一个节点,再计算与当前节点的差,同时更新最小值
时间复杂度:O(N)
空间复杂度:O(1)
func getMinimumDifference(_ root: TreeNode?) -> Int {
var pre = -1
var res = Int.max;
func find(_ root: TreeNode?) {
guard let root = root else { return }
// 左
find(root.left)
// 中序遍历
if pre == -1 {
pre = root.val
}else{
// 计算差值
res = min(res, root.val-pre)
pre = root.val
}
// 右
find(root.right)
}
find(root)
return res
}
五笔学习
小程序:XWubi
配合公众号文章一起食用,效果更佳~
五笔教程之字根解析(三)