【lc】 414第三大的数

发布时间 2023-08-25 11:15:10作者: BJFU-VTH

地址:

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

思路:

设置一个set用来存储数据就Ok了,set还能天然去重。

注意判断新数据是否在set中。

代码:

class Solution:
    def thirdMax(self, nums) -> int:
        tmp_ans = set()
        for num in nums:
            if num in tmp_ans:
                continue
            if tmp_ans.__len__() < 3:
                tmp_ans.add(num)
            else:
                mins = min(tmp_ans)
                if num > mins:
                    tmp_ans.remove(mins)
                    tmp_ans.add(num)
        print(tmp_ans)
        return min(tmp_ans) if len(tmp_ans) == 3 else max(tmp_ans)