（Python）剑指Offer40：最小的k个数

题目

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

示例

示例 1：
输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]

示例 2：
输入：arr = [0,1,2,1], k = 1
输出：[0]

限制

1. 0 <= k <= arr.length <= 10000
2. 0 <= arr[i] <= 10000

Python实现

思路 如下图所示：

class Solution(object):
    def getLeastNumbers(self, arr, k):
        """
        :type arr: List[int]
        :type k: int
        :rtype: List[int]
        """
        if k == 0:
            return list()
        self.randomized_selected(arr, 0, len(arr) - 1, k)
        return arr[:k]
    def partition(self, nums, l, r):  #快速排序
        pivot = nums[r]
        i = l - 1
        for j in range(l, r):
            if nums[j] <= pivot:
                i += 1
                nums[i], nums[j] = nums[j], nums[i]
        nums[i + 1], nums[r] = nums[r], nums[i + 1]
        return i + 1

    def randomized_partition(self, nums, l, r):
        i = random.randint(l, r)   #随机选取快排依据值
        nums[r], nums[i] = nums[i], nums[r]
        return self.partition(nums, l, r)

    def randomized_selected(self, arr, l, r, k):
        pos = self.randomized_partition(arr, l, r)
        num = pos - l + 1
        if k < num:  #若中位数左侧数值个数多于k
            self.randomized_selected(arr, l, pos - 1, k)
        elif k > num:  
            self.randomized_selected(arr, pos + 1, r, k - num)

【甜点来袭1】list中~的使用:
data = [1,2,3,4]
print(data[~0]
输出结果：4
list[~number]表示取list中倒数第number+1个数据

【甜点来袭2】<<、>>的使用:
print(7 >> 1 , 7 << 1)
输出结果：3 14
运算符>>表示右移1位（等同于‘//’即除以2），运算符<<左移1位（等同于乘以2）。

【甜点来袭3】BFPRT算法原理如下图所示：

参考资料

https://blog.csdn.net/LaoJiu_/article/details/54986553?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task