leetcode LCR 159. 库存管理 III（easy）（小林出品必属精品）

链接：力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

代码：

class Solution {
    public int[] inventoryManagement(int[] stock, int cnt) {
        //逐步找出 stock 数组的 0，stock.length-1 区间内最小的 cnt 个数，将其移动到数组前方
        qSelect(stock,0,stock.length-1,cnt);
        int[]ret=new int[cnt];
        //将前 cnt 个数据取出
        for(int i=0;i=r){
            return;
        }
        //随机选取一个基准元素 key ，将数组分为 3 部分
        int key=stock[new Random().nextInt(r-l+1)+l];

        int left=l-1,right=r+1,i=l;
        while(ikey 区间找最小的 cnt - a - b 个数,

题解：

        该题简单来说就是去获取数组中最小的 cnt 个数，解决该问题有很多的方法，可以将数组直接排好序，返回最小的 cnt 个数，时间复杂度为 O（nlogn）,也可以使用大根堆来辅助解决，时间复杂度也为O（nlogn），该题我们采用的是快速选择的方法，时间复杂度能达到 O（n）

         快速选择方法的核心就是快速排序的核心，要先随机取一个基准数据 key，按照基准数据 key 将数组划分为 3 个区间，分别是 > key , = key ,< key 的区间，这样就可以把小的数据往前放，根据每个区间的数据个数，判断最小的 cnt 个数在哪些区间，再去这些区间用同样的方式把小的数据往前放，当最小的 cnt 个数都被放到数组最前面时，直接返回前 cnt 个数据即可

        1.首先我们需要随机取一个基准数据 key，按照基准数据 key 将数组划分为 3 个区间，分别是 > key , = key ,< key 区间，该过程推荐看leetcode 75. 颜色分类（medium）（优质解法），这里就不赘述了

        2.将数组划分为 3 个区间以后如下图所示：

        其中 a，b，c 分别代表对应区间的数据个数，由图可知，a = left - L + 1 , b = right - left - 1 , c = R - right + 1 

        题目要求找最小的 cnt 个数，所以先在 < key 的区间找，当

        cnt < a 时 ，最小的 cnt 个数在 key 区间，直接去 < key 区间找最小的 cnt 个数，按照相同的方法进行划分，将小的数移至数组前方（递归）

        cnt = a 或 cnt <= a + b 时，代表 < key 区间的所有数据都是符合条件的，=key 区间的数据部分符合条件，但 = key 区间的数据都为 key，顺序不重要，这代表数组中前 cnt 个数据已经是满足条件的了，直接 return 即可

        不是上述情况的话，最小的 cnt 个数就会包含在 key 三个区间中，此时 key 区间中找到最小的 cnt - a - b 个数据即可，按照相同的方法进行划分，将小的数移至数组前方（递归）