yeonghwanjeon / coding_practice

0 stars 0 forks source link

[LeetCode]longest univalue path (완료) #58

Open yeonghwanjeon opened 5 years ago

yeonghwanjeon commented 5 years ago

class Solution(object): def longestUnivaluePath(self, root): """ :type root: TreeNode :rtype: int """ self.result = 0 def traverse(node) : if not node : return 0

        left, right = traverse(node.left), traverse(node.right)

        if node.left and node.val == node.left.val :
            left_len = left + 1
        else :
            left_len = 0

        if node.right and node.val == node.right.val :
            right_len = right + 1
        else :
            right_len = 0

        self.result = max(self.result, left_len + right_len)
        return max(left_len, right_len)

    traverse(root)
    return self.result