目录
一、问题
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和
k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和
k = 4
输出: 4
二、解题思路
分析:一种解题思路:简单考虑,就是个类似排序题,返回排序后(升序),返回倒数第k个数;
另一种解题思路:额外设置一个数组,用来存储前k大个数,思路和实现第二大、第三大个数的思路同理(特殊情况),这是普遍情况的一种实现而已,可以参考博文:https://blog.csdn.net/weixin_42521211/article/details/88364558,其中就是返回最大的三个数和最小的两个数。实现想法是一致的。
具体参考代码,一目了然。
思路和python具体实现
程序一:调用了sort()函数,竟然提交通过了,实现很简单:
class Solution(object):
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
nums.sort()
return nums[-k]
程序二:编写的可以存储前K大个值:
class Solution(object):
def findKthLargest(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
kmaxs = [float('-inf')]*k # 存储前k大个值
for i in range(len(nums)):
j = 0
while j<k:
if nums[i]> kmaxs[j]:
kmaxs[j+1:] = kmaxs[j:k-1] #有元素要改变时,先将待改变的当前元素值及之后元素依次向后赋值
kmaxs[j] = nums[i] # 然后修改要改变的当前元素
break
else:
j +=1
return kmaxs[k-1]
四、题外记
实现速度的快慢,应该是跟底层有关系,相对来说,还是第二个程序,更符合算法的实现逻辑。