码迷,mamicode.com
首页 > 其他好文 > 详细

783. 二叉搜索树节点最小距离

时间:2021-04-13 12:40:55      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:diff   lan   相同   com   记录   rom   lis   range   reverse   

题目

783. 二叉搜索树节点最小距离
给你一个二叉搜索树的根节点 root ,返回 树中任意两不同节点值之间的最小差值 。

注意:本题与 530:https://leetcode-cn.com/problems/minimum-absolute-difference-in-bst/ 相同

示例 1:
输入:root = [4,2,6,1,3]
输出:1

示例 2:
输入:root = [1,0,48,null,null,12,49]
输出:1
 
提示:
树中节点数目在范围 [2, 100] 内
0 <= Node.val <= 105

解法

  • 二叉搜索树经过中序遍历后所得到节点值组成的列表是升序的,只需要得到中序遍历列表,依次后面减前面,取最小即可。

    class Solution:
        def minDiffInBST(self, root: TreeNode) -> int:
            rootValList = []
            def reverse(root):
                if root:
                    reverse(root.left)
                    rootValList.append(root.val)
                    reverse(root.right)
            reverse(root)
            return min([rootValList[i+1] - rootValList[i] for i in range(len(rootValList) - 1)])
    
  • 当然也可以取一个变量记录前一节点的val,在列表中直接存储差值。这里为了不对结果产生影响,变量取值为-1e5。

    class Solution:
        def minDiffInBST(self, root: TreeNode) -> int:
            rootValDiffList = []
            self.p = -1e5
            def reverse(root):
                if root:
                    reverse(root.left)
                    rootValDiffList.append(root.val - self.p)
                    self.p = root.val
                    reverse(root.right)
            reverse(root)
            return min(rootValDiffList)
    
  • 更进一步,列表也可以不要,只需要记录前一节点的val,以及当前最小差值即可

    from math import inf
    
    class Solution:
        def minDiffInBST(self, root: TreeNode) -> int:
            self.minDiff = inf
            self.p = -inf
            def reverse(root):
                if root:
                    reverse(root.left)
                    self.minDiff = min(self.minDiff, root.val - self.p)
                    self.p = root.val
                    reverse(root.right)
            reverse(root)
            return self.minDiff
    

783. 二叉搜索树节点最小距离

标签:diff   lan   相同   com   记录   rom   lis   range   reverse   

原文地址:https://www.cnblogs.com/tensorzhang/p/14651275.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!