剑指Offer 40 最小的k个数

转载于 2019-06-13 09:22:00 发布 · 128 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/asenyang/p/11014210.html

本文介绍了一种高效算法，用于从一组整数中找出最小的K个数。通过使用Python的heapq库，实现了一个名为Solution的类，其中包含GetLeastNumbers_Solution方法，该方法接受整数列表和K作为参数，返回最小的K个整数。

最小的k个数

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

 1 import heapq
 2 # -*- coding:utf-8 -*-
 3 class Solution:
 4     def GetLeastNumbers_Solution(self, tinput, k):
 5         n = len(tinput)
 6         if k > n:
 7             return []
 8         result = heapq.nsmallest(k, tinput)
 9         return result
10         # write code here