687. 最长同值路径

合集 - LeetCode 树(11)

1.687. 最长同值路径2024-09-11

2.124. 二叉树中的最大路径和2024-09-113.2385. 感染二叉树需要的总时间2024-09-114.【模板题】- 144. 二叉树的前序遍历2024-09-125.【模板题】- 94. 二叉树的中序遍历2024-09-126.【模板题】- 145. 二叉树的后序遍历2024-09-127.872. 叶子相似的树2024-09-128.LCP 44. 开幕式焰火2024-09-129.404. 左叶子之和2024-09-1210.671. 二叉树中第二小的节点2024-09-1211.662. 二叉树最大宽度2024-09-24

收起

题目链接	687. 最长同值路径
思路	“树的直径”的简单变体
题解链接	【视频】彻底掌握直径 DP！从二叉树到一般树！（Python/Java/C++/Go）
关键点	注意维护当前状态：当左二子的值与本节点值不同时，要更新$v_l=0$
时间复杂度	$O(n)$
空间复杂度	$O(n)$

代码实现：

class Solution:
    def longestUnivaluePath(self, root: Optional[TreeNode]) -> int:
        answer = 0
        def dfs(node):
            if node is None:
                return -1
            l_len = dfs(node.left) + 1
            r_len = dfs(node.right) + 1
            if node.left and node.left.val != node.val:
                l_len = 0
            if node.right and node.right.val != node.val:
                r_len = 0
            nonlocal answer
            answer = max(answer, l_len + r_len)
            return max(l_len, r_len)
        
        dfs(root)
        return answer