给定一个非空数组,返回此数组中第三大的数。如果不存在,则返回数组中最大的数。要求算法时间复杂度必须是O(n)。
思路:
1、当输入数组中有两个元素,返回较大的
2、为避免重复元素的影响,对输入数组去重再做比较。如果原数组([1,1,2])去重后,新数组只有两个元素([1,2]),则返回较大的
3、取新数组的最大值,删掉。再取一次(第二大),删掉。此时数组中的最大值即为第三大的值
class Solution:
def thirdMax(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums) == 2:#如果数组中只有两个元素,返回大的
return max(nums)
newNums = set(nums) #set删除重复元素
if len(newNums) == 2: #去重后 数组中有两个元素 返回大的
return max(newNums)
elif len(newNums) == 1:#去重后 数组中只有一个元素 返回
return nums[0]
#两次删除最大值,再取最大值即为第三大的值
max1 = max(newNums)
newNums.remove(max1)
max2 = max(newNums)
newNums.remove(max2)
max3 = max(newNums)
return max3