劍指offer面試題40. 最小的k個數(sort函數)

題目描述

輸入整數數組 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]:
		arr.sort()
		return arr[:k]

複雜度

sort()函數使用 Timsort 方法進行排序,時間複雜度O(NlogN)