lintcode 题1

在数组中找到第k大的元素 样例 给出数组 [9,3,2,4,8],第三大的元素是 4 给出数组 [1,2,3,4,5],第一大的元素是 5,第二大的元素是 4,第三大的元素是 3,以此类推 挑战 要求时间复杂度为O(n),空间复杂度为O(1),设函数为 find(n,nums)nums为给定数组,n是指找到第n大的元素
此题可以采用快速选择来解。
找到数组nums[]中间元素nums[mid],然后创建两个数组big[] 和small[]分别存放大于nums[mid]和小于nums[mid]的元素然后重复以下过程:
1.如果big数组的长度等于n-1,那么说明nums[mid]为第n大的元素,return nums[mid]
2如果big数组的长度大于n-1,那么说明第n大的元素在数组big中,执行 return find(n,big)
3.如果big数组长度小于n-1,那么说明第n大的元素在数组small中,执行return find(n-len(big)-1-num,small)
这里的num是用来处理重复的nums[mid]的

def find(n,nums):
    if n > len(nums) or n < 1 or nums == None:
            return None
        mid = len(nums)/2
        big = []
        small = []
        num = 0
        for i in range(len(nums)):
            if nums[i] > nums[mid]:
                big.append(nums[i])
            if nums[i] < nums[mid]:
                small.append(nums[i])
            if nums[i] == nums[mid]:
                num += 1
        num -= 1
        if len(big) == n - 1:
            return nums[mid]
        if len(big) > n - 1:
            return find(n,big)
        if len(big) < n - 1:
            if len(big) + num >= n-1:
                return nums[mid]
            else:

                return find(n-len(big)-1-num,small)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值