[Leetcode] 414. 第三大的数 Python3

 给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是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