题目描述
输入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:]