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

2021/03/06 leetcode 共 649 字,约 2 分钟

title

problem link

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

(如果 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

提示:

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

solution

class Solution:
    def maxAncestorDiff(self, root: TreeNode) -> int:
        ans = 0
        if not root:
            return 0
        def dfs(node, minn, maxn):
            nonlocal ans
            if not node:
                return 0
            minn = min(minn, node.val)
            maxn = max(maxn, node.val)
            # 可以到叶子节点再判断
            ans = max(ans, abs(maxn - minn))
            dfs(node.left, minn, maxn)
            dfs(node.right, minn, maxn)
        dfs(root, float('inf'), float('-inf'))
        return ans

Search

    Table of Contents