暴力解法:
先去重,然后降序排序,去重排序后的数组的长度大于等于3的时候,取下标为2的即为结果值,当去重后的数组长度小于3,取最大值即可
class Solution:
def thirdMax(self, nums: List[int]) -> int:
nums = list(set(nums))
nums.sort(reverse=True)
if len(nums) >= 3:
return nums[2]
else:
return max(nums)
题目中还有一个要求:
进阶:你能设计一个时间复杂度 O(n)
的解决方案吗?
使用堆(特别是最小堆)来解决这个问题,可以帮助我们保持跟踪数组中最大的三个数。在Python中,我们可以使用标准库heapq
里的最小堆实现。关键思路是维护一个大小为3的最小堆,这样堆顶始终是这个堆中最小的数,当堆的大小超过3时,我们可以弹出最小的数,最后堆中剩下的就是数组中最大的三个数。如果堆的大小小于3,说明不存在第三大的数,则返回堆中的最大数。
import heapq
class Solution:
def thirdMax(self, nums: List[int]) -> int:
heap = []
nums = set(nums)
for num in nums:
heapq.heappush(heap, num)
if len(heap) > 3:
heapq.heappop(heap)
return heap[0] if len(heap) == 3 else max(heap)