题目:给定二叉树的根节点 root,找出存在于 不同 节点 A 和 B 之间的最大值 V,其中 V = |A.val - B.val|,且 A 是 B 的祖先。
(如果 A 的任何子节点之一为 B,或者 A 的任何子节点是 B 的祖先,那么我们认为 A 是 B 的祖先)
示例 1:
输入:root = [8,3,10,1,6,null,14,null,null,4,7,13]
输出:7
解释:
我们有大量的节点与其祖先的差值,其中一些如下:
|8 - 3| = 5
|3 - 7| = 4
|8 - 1| = 7
|10 - 13| = 3
在所有可能的差值中,最大值 7 由 |8 - 1| = 7 得出。
思路:最大的绝对值只能出现在子节点的最大值或者最小值。
代码:(low low 代码,计算以每个节点为根的子树的最大值和最小值,使用ans计算最大的绝对值)
# 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 maxAncestorDiff(self, root: Optional[TreeNode]) -> int:
def dfs(root):
val = root.val
mx = val
mi = val
if root.left is not None:
mx1, mi1 = dfs(root.left)
mx, mi = max(mx, mx1), min(mi, mi1)
if root.right is not None:
mx1, mi1 = dfs(root.right)
mx, mi = max(mx, mx1), min(mi, mi1)
self.ans = max(max(abs(val - mx), abs(val-mi)), self.ans)
return mx, mi
self.ans = 0
dfs(root)
return self.ans