【LeetCode】687. 最长同值路径

最新推荐文章于 2024-06-12 10:00:23 发布

pass night

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量421

点赞数

文章标签： leetcode 算法深度优先职场和发展

本文链接：https://blog.csdn.net/apple_50661801/article/details/126655813

版权

题目

687. 最长同值路径

给定一个二叉树的 root ，返回 最长的路径的长度 ，这个路径中的 每个节点具有相同值 。这条路径可以经过也可以不经过根节点。

两个节点之间的路径长度 由它们之间的边数表示。

示例 1:

在这里插入图片描述

输入：root = [5,4,5,1,1,5]
输出：2

示例 2:

在这里插入图片描述

输入：root = [1,4,5,4,4,5]
输出：2

提示:

树的节点数的范围是 [0, 104]
-1000 <= Node.val <= 1000
树的深度将不超过 1000

思路

动态规划， $d p [n o d e]$ 表示以node为根，左右子树中最长相同路径长度，则
$\left\{\begin{aligned} &0 &node.left.val \ne node.val \land node.right.val \ne node.val \\ &dp[node.left]+1 & node.val = node.left.val \land node.val \ne node.right.val\\ &dp[node.right]+1 & node.val = node.right.val \land node.val \ne node.left.val\\ &\max(dp[node.left]+1, dp[node.right]+1) & node.left.val=node.right.val=node.val \end{aligned} \right.$
结果值则为递归过程中最大的 $d p [n o d e . l e f t] + n o d e . r i g h t$

代码

# 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 longestUnivaluePath(self, root: Optional[TreeNode]) -> int:
        ret = 0
        def dfs(node: Optional[TreeNode]) -> int:
            nonlocal ret
            if not node:
                return 0
            left = dfs(node.left)
            right = dfs(node.right)
            leftLength = left + 1 if node.left and node.val == node.left.val else 0
            rightLength = right + 1 if node.right and node.val == node.right.val else 0
            ret = max(ret, leftLength + rightLength)
            return max(leftLength, rightLength)
        dfs(root)
        return ret