LeetCode-Python-5030. 节点与其祖先之间的最大差值

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

（如果 A 的任何子节点之一为 B，或者 A 的任何子节点是 B 的祖先，那么我们认为 A 是 B 的祖先）

示例：

输入：[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 到 5000 之间。
每个节点的值介于 0 到 100000 之间。

思路：

DFS + 回溯找出每条路径，

对于每个node，都计算一下它所在路径上的最大值和最小值和它的val的差值的绝对值，

然后判断要不要刷新结果。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def maxAncestorDiff(self, root):
        """
        :type root: TreeNode
        :rtype: 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

LeetCode-Python-5030. 节点与其祖先之间的最大差值

相关推荐