题解 | #最小的K个数#

直接利用python的heapq数据结构就行，而且里面还有个 heapq.nsmallest(k , input)函数，一步搞定，建堆的时间复杂度是O(nlogn)的。

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param input int整型一维数组 
# @param k int整型 
# @return int整型一维数组
#
import heapq

class Solution:
    def GetLeastNumbers_Solution(self , input, k: int) :
        # write code here
        # 构建最小堆，顺序返回前四个即可
        heapq.heapify(input)
        print(input)
        return heapq.nsmallest(k , input)

input = [4,5,1,6,2,7,3,8]
k = 4
print(Solution().GetLeastNumbers_Solution(input , k))