最小的K个数

题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
方法:
用最大堆实现,先建立一个k大堆,然后不断比较维护一个k大堆。最后再运用K大堆对k个树从小到大排列输出。

class Solution:
    def duipai(self,i,HeapList,ListLen):
        HeapList[0] = HeapList[i]
        j = i*2
        while(j <= ListLen):
            if j < ListLen and HeapList[j] < HeapList[j+1]:
                j = j+1
            if HeapList[0] >= HeapList[j]:
                break
            HeapList[i] = HeapList[j]
            i = j
            j = j*2
        HeapList[i] = HeapList[0]
    def GetLeastNumbers_Solution(self, tinput,k):
        # write code here
        if len(tinput) < k or k == 0:
            return []
        '''取前K个数进行堆排'''
        list2 = tinput[0:k]
        list2 = [0] + list2
        for i in range(k // 2,0,-1):
            self.duipai(i,list2,k)
        '''从第k+1个数开始与堆顶比较'''
        for value in tinput[k:]:
            if value < list2[1]:
                list2[1] = value
                self.duipai(1,list2,k)
        '''最后K个值堆排实现从小到大排列'''
        for m in range(k,1,-1):
            list2[m],list2[1] = list2[1],list2[m]
            self.duipai(1,list2,m -1)
        return list2[1:]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值