剑指 Offer 40. zui-xiao-de-kge-shu-lcof 最小的k个数 python3

时间：2021-03-30

题目地址：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/

题目难度：Easy

题目描述：

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

示例 1：

输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]
示例 2：

输入：arr = [0,1,2,1], k = 1
输出：[0]

---

思路1:排序

代码段1：通过

# -*- coding:utf-8 -*-
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        if k > len(tinput):
            return []
        tinput.sort()
        return tinput[:k]

总结：

1. 简单题就排序吧

---

思路2:堆

代码段2：通过

class Solution:
    def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
        if k > len(arr):
            return []
        res = []
        heapq.heapify(arr)
        for i in range(k): 
            res.append(heapq.heappop(arr))
        return res

总结：

1. python的是个最小堆，永远弹出一个值最小的数