题目
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