剑指Offer 40 最小的k个数

最小的k个数

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

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

 

Java版代码，leetcode地址：

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        Arrays.sort(arr);
        int[] result = Arrays.copyOfRange(arr, 0, k);
        return result;
    }
}