最小的K个数

最小的K个数-牛客

思路

• 堆排序：优先队列
  每次插入元素时间复杂度都是O(logn)，每次取出堆顶元素都是直接取出。
  1：利用input数组中前k个元素，构建一个大小为k的大顶堆，堆顶为这k个元素的最大值，python中heapq默认是小根堆，为了构造大根堆，需要对每个元素*-1。
  2：对于后续的元素，依次比较其与堆顶的大小，若是比堆顶小，则堆顶弹出，再将新数加入堆中，直至数组结束，保证堆中的k个最小。
  3：最后将堆顶依次弹出即是最小的k个数

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param input int整型一维数组 
# @param k int整型 
# @return int整型一维数组
#
import heapq
class Solution:
    def GetLeastNumbers_Solution(self , input: List[int], k: int) -> List[int]:
        # 堆排序-优先队列
        #求最小值，用大顶堆，每次与堆顶元素比较，比堆顶小，弹出堆顶，弹入较小值
        n=len(input)
        res=[]
        if n==0 or k==0:
            return res
        if n>=k and k!=0:
            #heapq默认小根堆，若要构造大根堆，每次输入要乘-1
            pq=[]
            for i in range(k):
                #构建一个大小为k的堆
                heapq.heappush(pq,(-1*input[i]))
            for i in range(k, n):
                #较小元素入堆
                if (-1 * pq[0]) > input[i]:
                    #删除最小值并添加新元素
                    heapq.heapreplace(pq, (-1 * input[i]))
            #取出堆中元素
            for i in range(k):
                res.append(-1*pq[0])
                heapq.heappop(pq)
        return res