题目
分析
题目的意思很明确。这是典型的topk问题,可以维护一个最小堆,每次从堆的最顶端取数。
建堆的过程是一个上浮的过程,每次把新的数放在最后面,依次找K/2父节点的位置,判断父节点和子节点 谁小,子节点小和父节点交换并且继续上浮,否则停止上浮。
输出最小值的过程是一个下潜的过程,把根取出,把最后一位补在第一位。然后让根的数值下沉,每次通过k2和k2+1两个位置的子节点值和父节点值对比,如果父节点大则继续下沉,否则停止下沉。
整道题就是先建堆,再取出堆的最小值,更新堆的过程。
python代码
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
# write code here
if k>len(tinput): return []
heap = Heap()
for i in tinput:
heap.insert(i)
res = []
for i in range(min(k,len(tinput))):
res.append(heap.removeMin())
return res
class Heap:
def __init__(self):
self.item = [0]
self.N = 0
def swim(self,K):
while K > 1:
f_index = K>>1
if self.item[f_index]>self.item[K]:
self.item[f_index],self.item[K]=self.item[K],self.item[f_index]
K = f_index
else:
break
def sink(self,K):
while 2*K+1<=self.N:
child_1 = 2*K
child_2 = 2*K+1
child = child_2 if self.item[child_1]>self.item[child_2] else child_1
if self.item[K]>self.item[child]:
self.item[child],self.item[K]=self.item[K],self.item[child]
K = child
else:
break
def insert(self,num):
if self.N>0:
self.item.append(num)
self.N+=1
self.swim(self.N)
else:
self.item.append(num)
self.N+=1
def removeMin(self):
if self.N>1:
self.item[1],self.item[-1] = self.item[-1],self.item[1]
self.N-=1
res = self.item.pop(-1)
self.sink(1)
return res
elif self.N==1:
self.N-=1
res = self.item.pop(-1)
return res
else:
return None