【剑指offer】29 最小的K个数

题目

分析

题目的意思很明确。这是典型的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