python
class Solution:
def thirdMax(self, nums: List[int]) -> int:
nums.sort()
# print(nums)
max_nums = nums[-1]
count = 1
i =len(nums)-1
while(i>0):
if(nums[i]!=nums[i-1]):
count +=1
if(count==3):
break
i -=1
# print(count,i)
if(count==3):
return nums[i-1]
else:
return max_nums