1250. 第三大的数

1250. 第三大的数

中文English

给定一个非空数组，返回此数组中第三大的数。如果不存在，则返回数组中最大的数。要求算法时间复杂度必须是O(n)。

样例

样例 1:

输入: num = [3, 2, 1]
输出: 1
解释: 第三大的数是 1.

样例 2:

输入: num = [1, 2]
输出: 2
解释: 第三大的数不存在, 所以返回最大的数 2 .

样例 3:

输入: num = [2, 2, 3, 1]
输出: 1
解释: 注意，要求返回第三大的数，是指第三大且唯一出现的数。
存在两个值为2的数，它们都排第二。

class Solution:
    """
    @param nums: the array
    @return: the third maximum number in this array
    """
    '''
    大致思路:
    1.首先进行排序，初始化count，然后循环nums，如果下一个值与上一个不相等则count += 1，直到count == 3，则返回。否则返回最大值nums[0]
    '''
    def thirdMax(self,nums):
        nums = sorted(nums)[::-1]
        count = 0
        for i in range(len(nums)-1):
            if nums[i+1] != nums[i]:
                count += 1
                if count == 2:
                    return nums[i+1]
        return nums[0]