sorted不符合题意,但是也通过了
class Solution:
def thirdMax(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
A = sorted(list(set(nums)))
if len(A) < 3:
return max(A)
else:
A.reverse()
return A[2]
正解,思路很棒!
class Solution(object):
def thirdMax(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
a = set(nums)
if len(a) < 3:
return max(a)
else:
a.remove(max(a))
a.remove(max(a))
return max(a)