Skip to content

1026. 节点与其祖先之间的最大差值

题目

给定二叉树的根节点 root,找出存在于 不同 节点 AB 之间的最大值 V,其中 V = |A.val - B.val|,且 AB 的祖先。

(如果 A 的任何子节点之一为 B,或者 A 的任何子节点是 B 的祖先,那么我们认为 A 是 B 的祖先)

示例 1:

img

输入: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:

img

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

提示:

  • 树中的节点数在 25000 之间。
  • 0 <= Node.val <= 10^5

解答

代码

等价于求出树中的最大值和最小值:

python
class Solution:
    def maxAncestorDiff(self, root: Optional[TreeNode]) -> int:
        min_tree = root.val
        max_tree = root.val
        ans = 0

        def dfs(node):
            if node is None:
                return

            nonlocal min_tree
            nonlocal max_tree
            nonlocal ans

            t1 = min_tree
            t2 = max_tree

            min_tree = min(min_tree, node.val)
            max_tree = max(max_tree, node.val)
            ans = max(ans, max_tree - min_tree)

            dfs(node.left)
            dfs(node.right)

            min_tree = t1
            max_tree = t2
        
        dfs(root)

        return ans

不用恢复现场的做法,把 min, max 当作 形参 传入递归函数:

python
class Solution:
    def maxAncestorDiff(self, root: Optional[TreeNode]) -> int:
        ans = 0
        
        def dfs(node: Optional[TreeNode], mn: int, mx: int) -> None:
            if node is None:
                return

            mn = min(mn, node.val)
            mx = max(mx, node.val)
            
            nonlocal ans
            ans = max(ans, node.val - mn, mx - node.val)
            
            dfs(node.left, mn, mx)
            dfs(node.right, mn, mx)
        
        dfs(root, root.val, root.val)
        
        return ans

Released under the MIT License.