Leetcode 刷题- 414第三大的数

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

示例 1:

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

示例 2:

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

示例 3:

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

解题思路：

1、列表长度小于3时， 直接如返回最大值
2、列表大于等于3时，对列表进行倒排，即从大到小排列， 进行循环遍历， 值等于最大值时，跳过； 不等于时，最大值变更，并且计数+1， 判断i 是否等于3，等于则直接返回； 循环结束后，计数小于3，则返回最大值，  代码如下：

代码 版本1

class Solution(object):
    def thirdMax(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        
        max_ = max(nums)
        if len(nums) < 3:
            return max_
        nums.sort(reverse=True)
        i = 1
        for num in nums:
            if num == max_:
                continue
            else:
                i +=1
                max_ = num
            if i == 3:
                return num
        if i < 3:
            return max(nums)

代码2 优化：
思路：
考虑到列表中存在重复元素，可以优先进行去重操作， 去重后，再进行排序，再判断 列表长度；
小于3， 则直接返回 最大值； 大于等于3， 则返回 第3个数，

class Solution(object):
    def thirdMax(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        nums = list(set(nums))
        nums.sort()
        if len(nums) <3:
            return nums[-1] 
        else:
            return nums[-3]