top k问题python解

题目来源:Leetcode 215 数组中第K个最大的元素

题目描述:

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2

输出: 5

1.排序方法

复杂度为O(nlogn),可以直接调用python的排序函数做(用时52ms):

class Solution:
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        nums.sort()
        return nums[-k]
        

自己手写快排(用时284ms,咋这么慢。。。):

class Solution:
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        import random
        self.quickSort(nums, 0, len(nums) - 1)
        return nums[-k]
    
    
    def partation(self, nums, l, r):
        """随机partation"""
        change_index = random.randint(l, r)
        nums[change_index], nums[r] = nums[r], nums[change_index]
        num = nums[r]
        less_index = l - 1
        more_index = r + 1
        p = l
        while p < more_index:
            if nums[p] < num:
                less_index += 1
                nums[p], nums[less_index] = nums[less_index], nums[p]
                p +&#
  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值