美文网首页
LeetCode 414. Third Maximum Numb

LeetCode 414. Third Maximum Numb

作者: singed | 来源:发表于2018-09-02 14:42 被阅读0次

链接

https://leetcode-cn.com/problems/third-maximum-number/description/

要求

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

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

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

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

思路

如果set后长度<3,则返回最后一个数
如果set后长度>=3,则返回倒数第三个数

代码

执行用时:28 ms

class Solution(object):
    def thirdMax(self, nums):
        if len(set(nums)) < 3:
            return sorted(list(set(nums)))[-1]
        else:
            return sorted(list(set(nums)))[-3]
image.png

相关文章

网友评论

      本文标题:LeetCode 414. Third Maximum Numb

      本文链接:https://www.haomeiwen.com/subject/vddmwftx.html