Leetcode每日一题2021.2.25第687题：最长同值路径

题目描述

示例

思路

我们可以将任何路径看作是两个从其根延伸出的箭头，且最多只有两个箭头。
对于每个节点，我们想知道 向左延伸的最长箭头和向右延伸的最长箭头是什么？我们可以用递归来解决这个问题。

算法

令 arrow_length(node) 为从节点 node 延伸出的最长箭头的长度。如果 node.left 存在且与节点 node 具有相同的值，则该值就会是 1+ arrow_length(node.left)。在 node.right 存在的情况下也是一样。
当我们计算箭头长度时，候选答案 self.ans 将是该节点在两个方向上的箭头之和。我们将这些 self.ans 记录下来，并返回最佳答案。

代码

# 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: TreeNode) -> int:
        self.ans = 0
        
        #  arrow_length(node)函数返回 从节点node延伸出的最长箭头的长度
        def arrow_length(node):
            if not node:
                return 0
            left_length = arrow_length(node.left)
            right_length = arrow_length(node.right)
            left_arrow = right_arrow = 0
            if node.left and node.left.val == node.val:
                left_arrow = left_length + 1
            if node.right and node.right.val == node.val:
                right_arrow = right_length + 1
            #  候选答案为 该节点在两个方向上的箭头之和
            self.ans = max(self.ans, left_arrow + right_arrow)
            return max(left_arrow, right_arrow)
        
        arrow_length(root)
        return self.ans