Leetcode刷题记录 day15

最新推荐文章于 2024-09-14 18:52:56 发布

wu-deng

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量371

点赞数 5

分类专栏： leetcode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_62164397/article/details/137398560

版权

leetcode 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

每日一题节点与其祖先之间的最大差值

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

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

示例 1：

输入：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 得出。

解题思路

根据示例1的输入我们可以知道，输入的二叉树使用列表表示，除根节点外，第i个节点的左节点索引为2i+1，右节点为2（i+1），同时，第i个节点的父节点索引为(i-1) // 2，我们可以从根节点开始，用类似深度遍历的形式获取每个从根节点到叶子节点的子树，如(8,3,1)、(8,3,6,4)并取出其中的最大最小值，更新最大差值，即为所求。

代码实现

class Solution:
    def maxAncestorDiff(self, root: Optional[TreeNode]) -> int:
        def dfs(tree, mi, ma):
            if tree == None:
                return 0
            diff = max(abs(tree.val - mi), abs(tree.val - ma))
            mi = min(tree.val, mi)
            ma = max(tree.val, ma)
            diff = max(diff, dfs(tree.left, mi, ma))
            diff = max(diff, dfs(tree.right, mi, ma))
            return diff
        return dfs(root, root.val, root.val)