且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

[leetcode/lintcode 题解]算法面试真题详解:二叉搜索树结点最小距离

更新时间:2022-01-15 04:35:02

描述
给定一个二叉搜索树的根结点 root, 返回树中任意两节点的差的最小值。

  • 二叉树的大小范围在 2 到 100。
  • 二叉树总是有效的,每个节点的值都是整数,且不重复。

在线评测地址:领扣题库官网

样例1
输入: root = {4,2,6,1,3}
输出: 1
解释:
注意,root是树结点对象(TreeNode object),而不是数组。

给定的树 [4,2,6,1,3,null,null] 可表示为下图:

          4
        /   \
      2      6
     / \    
    1   3  

最小的差值是 1, 它是节点1和节点2的差值, 也是节点3和节点2的差值。
样例2
输入: root = {2,1}
输出: 1
解释:
注意,root是树结点对象(TreeNode object),而不是数组。

给定的树 {2,1} 可表示为下图:

      2
     / 
    1 

最小的差值是 1, 它是节点1和节点2的差值。

解题思路
二叉搜索树,根据树的性质,知道root的右子树都是大于它的,左子树都是小于它的。
那么如果做中序遍历,标准的做法是得到一个递增的序列。
我们就先遍历左根,节点,右根,这样就会得到一个递增的序列。
然后对这个序列相邻相减,取最小值即可。 实现时,可以优化掉这个序列。
在遍历时记录上一个访问的节点值,和当前节点相减,记录下最小值即可。

源代码

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""

class Solution:
    """
    @param root:  a Binary Search Tree (BST) with the root node
    @return: the minimum difference
    """
    pre = -float('inf')
    res = float('inf')

    def minDiffInBST(self, root):
        # Write your code here.
    
        if root.left:
            self.minDiffInBST(root.left)
        self.res = min(self.res, root.val - self.pre)
        self.pre = root.val
        if root.right:
            self.minDiffInBST(root.right)
        return self.res

更多题解参考:九章官网solution