【二叉树】二叉搜索树的最小绝对差

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
配合公众号文章一起食用,效果更佳~
五笔教程之字根解析(三)
请添加图片描述


Guess you like

Origin blog.csdn.net/xjh093/article/details/121386074