1026. 节点与其祖先之间的最大差值-CSDN博客

本文链接：https://blog.csdn.net/u013243296/article/details/115374069

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

原始题目链接：https://leetcode-cn.com/problems/maximum-difference-between-node-and-ancestor/

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

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

在这里插入图片描述

在这里插入图片描述
解题思路：

根据题意，在遍历树中节点的同时，记录更新比较树中节点的最大值和最小值，所以使用先根遍历序列的方式遍历二叉树。另外题目要求节点不能横跨，所以在遍历到每个叶子节点结束之后，和当前的最大差值进行比较，更新最大差值。具体实现看代码及注释。

代码实现：

# 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 __init__(self):
        self.ans = float('-inf')
        self.max_value = float('-inf')
        self.min_value = float('inf')

    # 定义递归函数：深度优先遍历树中的节点，在递归遍历节点的同时，更新最大值和最小值，所以使用先根遍历的方式
    # 因为题目要求不能夸左右子树，所以在遍历到叶子节点的时候，比较当前最大差值和答案的值，更新答案
    def dfs(self, root, max_value, min_value):
        # 叶子节点都遍历完了，更新答案
        if not root:
            self.ans = max(max_value - min_value, self.ans)
        else:
            # 先根遍历
            max_value, min_value = max(root.val, max_value), min(root.val, min_value)
            self.dfs(root.left, max_value, min_value)
            self.dfs(root.right, max_value, min_value)

    def maxAncestorDiff(self, root: TreeNode) -> int:
        self.dfs(root, self.max_value, self.min_value)
        return self.ans