题目在这:https://leetcode-cn.com/problems/minimum-absolute-difference-in-bst/
思路在这:
这题有个小坑。我上来直接给所有节点全推进数组了,然后排序了数组,取了一号位减零号位的差值,发现不对。
题目中要求的是任意两个节点直接的差值绝对值的最小值。
所以:
- 用先序或者中序后序等将所有节点的值推入数组中。
- 对数组进行排序
- 遍历整个数组,计算出来每两个数直接的差值。并记录最小值输出、
完整代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def getMinimumDifference(self, root: TreeNode) -> int:
res = []
def dfs(root):
if not root:
return
res.append(root.val)
dfs(root.left)
dfs(root.right)
dfs(root)
res.sort()
temp = res[1]-res[0]
for i in range(len(res)-1):
temp = min(temp,res[i+1]-res[i])
return temp