剑指Offer 40.最小的k个数（Python）

题目描述

输入n个整数，找出其中最小的k个数。
示例：

输如4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

---

解题思路

排序

直接将n个整数排序，返回前k个数。时间复杂度是$O(nlogn)$。

class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        if not tinput: return []
        
        sorted_tinput = self.quickSort(tinput)
        
        return sorted_tinput[:k] if k <= len(tinput) else []
        
        
    def quickSort(self,nums):
        if not nums: return []
        
        mid = nums[len(nums)//2]
        left, right = [], []
        
        for num in nums:
            if num < mid:
                left.append(num)
            if num > mid:
                right.append(num)
        
        return self.quickSort(left) + [mid] + self.quickSort(right)

---