leetcode刷题笔记25

面试题40. 最小的k个数

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

class Solution:
    def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
    	if not arr:
    		return None
    	nums = [0]*10000
    	for i in arr:
    		nums[i] += 1
    	res = []
    	i = 0
    	while len(res)<k:
      		if nums[i]>0:
    			for _ in range(nums[i]):
    				res.append(i)
       		i += 1
    	return res[:k]