LC 1026. 节点与其祖先之间的最大差值
题目描述
给定二叉树的根节点 root,找出存在于 不同 节点 A 和 B 之间的最大值 V,其中 V = |A.val - B.val|,且 A 是 B 的祖先。
(如果 A 的任何子节点之一为 B,或者 A 的任何子节点是 B 的祖先,那么我们认为 A 是 B 的祖先)
树中的节点数在 2 到 5000 之间。
0 <= Node.val <= 10^5
输入: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 得出。
思路
递归
|A-B|的最大值,对于A,|A-B|的最大值只可能是max(|A-min(B)| , |A-max(B)|)
有两种写法
方法1:自顶向下
方法2:自底向上
AC代码
class Solution:
def maxAncestorDiff(self, root: Optional[TreeNode]) -> int:
ans = 0
def dfs(root,ma,mi):
nonlocal ans
if root.left:
dfs(root.left,max(ma,root.left.val),min(mi,root.left.val))
ans = max(ans,abs(root.left.val-ma),abs(root.left.val-mi))
if root.right:
dfs(root.right,max(ma,root.right.val),min(mi,root.right.val))
ans = max(ans,abs(root.right.val-ma),abs(root.right.val-mi))
dfs(root,root.val,root.val)
return ans