标签: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
标签:diff lan 相同 com 记录 rom lis range reverse
原文地址:https://www.cnblogs.com/tensorzhang/p/14651275.html