亚马逊面试题：二叉搜索树结点最小距离

LintCode

于 2021-05-18 09:45:58 发布

阅读量127

点赞数

分类专栏： leetcode/lintcode每天一题文章标签：二叉树算法数据结构 leetcode python

本文链接：https://blog.csdn.net/LintCode/article/details/116976399

版权

leetcode/lintcode每天一题专栏收录该内容

107 篇文章 5 订阅

订阅专栏

这道题目要求在给定的二叉搜索树中找到任意两个节点间的最小差值。通过中序遍历，我们可以得到一个递增序列，然后在遍历过程中计算相邻节点的差值并记录最小值。最终输出这个最小差值。

摘要由CSDN通过智能技术生成

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

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

在线评测地址

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

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

最小的差值是 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

更多题解参考