LeetCode 414. 第三大的数（python）

题目链接

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

示例 1:

输入: [3, 2, 1]

输出: 1

解释: 第三大的数是 1.
示例 2:

输入: [1, 2]

输出: 2

解释: 第三大的数不存在, 所以返回最大的数 2 .
示例 3:

输入: [2, 2, 3, 1]

输出: 1

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

解题思路：
python中用float(“inf”), float("-inf")分别表示正负无穷
分别设m,mm,mmm,为负无穷，用来表示第一大、第二大、第三大的数字
遍历nums，
若当前数字大于第一大，则将当前数字设为第一大，并将原来的第一大、第二大分别改为第二大、第三大
以此类推

class Solution(object):
    def thirdMax(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        mmm,mm,m=float('-inf'), float('-inf'),float('-inf')
        for i in nums:
            if i>m:
                m,mm,mmm=i,m,mm
            elif m>i>mm:
                mm,mmm=i,mm
            elif mm>i>mmm:
                mmm=i
        return m if mmm== float('-inf') else mmm