1026：节点与其祖先之间的最大差值（1446 分）

力扣第 132 场周赛第 2 题

题目

给定二叉树的根节点 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 得出。

示例 2：

输入：root = [1,null,2,null,0,3]
输出：3

提示：

• 树中的节点数在 2 到 5000 之间。
• 0 <= Node.val <= 105

分析

遍历时维护路径的最大和最小值即可。

解答

1
2
3
4
5
6
7
8
9

def maxAncestorDiff(self, root: Optional[TreeNode]) -> int:
	res, stack = -inf, [(root,-inf,inf)]
	while stack:
		u,ma,mi = stack.pop()
		res = max(res,ma-u.val,u.val-mi)
		ma = max(ma,u.val)
		mi = min(mi,u.val)
		stack.extend((v,ma,mi) for v in [u.right,u.left] if v)
	return res

44 ms