LeetCode215. python实现:数组中的第K个最大元素问题☆☆

目录

一、问题

二、解题思路

三、python具体实现

四、题外记


一、问题

在未排序的数组中找到第 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]

四、题外记

      实现速度的快慢,应该是跟底层有关系,相对来说,还是第二个程序,更符合算法的实现逻辑。


版权声明:本文为weixin_42521211原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。