[LeetCode] Third Maximum Number (Python)

 

 

class Solution:
    def thirdMax(self, nums: List[int]) -> int:
        nums = set(nums)
        
        nums = sorted(nums, reverse=True)
        
        if len(nums) < 3:
            answer = nums[0]
        else:
            answer = nums[2]
        return answer