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

104 阅读1分钟

要求

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

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

示例 1:

image.png

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

image.png

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

提示:

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

核心代码

# 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:
        self.s = 0

        def find(node,path):
            if not node:
                return
            tmp = max(abs(max(path) - node.val),abs(min(path) - node.val))
            self.s = max(self.s,tmp)
            path.append(node.val)
            find(node.left,path)
            find(node.right,path)
            path.pop()
        
        find(root,[root.val])
        return self.s

image.png

解题思路:DFS + 回溯找出每条路径,对于每个node,都计算一下它所在路径上的最大值和最小值和它的val的差值的绝对值,然后判断要不要刷新结果。