剑指offer：最小的K个数

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

示例1

输入

[4,5,1,6,2,7,3,8],4

返回值

[1,2,3,4]

 

思路：使用堆排序实现

 

解法1：自己构建堆

 

 

解法2：使用python内置的sorted函数实现：sorted函数是不稳定的排序方法

# -*- coding:utf-8 -*-
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        if(k>len(tinput)):return []
        return sorted(tinput)[:k]

 

解法3：使用python内置的堆heapq实现

# -*- coding:utf-8 -*-
import heapq
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        if(k>len(tinput)):return []
        return heapq.nsmallest(k, tinput)

 

heapq堆的一些操作可见：https://www.jianshu.com/p/801318c77ab5