剑指 Offer 40 题 最小的 K个数

深入理解 JVM 原理
深入理解 JVM 原理
wangxiaoming

输入整数数组 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]

限制:

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

AC 通过,主要是用快排思想一次 partion 之后, 左边的比 哨兵小,右边的比 哨兵大。

public class JianZhi40 {

    public int[] getLeastNumbers(int[] arr, int k) {
        if(k == 0) return new int[]{};
        int len = arr.length -1 ;
        findK(arr,0,len,k);
        int[] nums = new int[k];
        for(int i = 0; i < k;i++){
            nums[i] = arr[i];
        }
        return nums;
    }
    /**
     * @param arr
     * @param i
     * @param len
     * @param k 
     * @return
     */
    private void findK(int[] arr, int start, int end, int k) {
        if (start < end) {
            int pos = partion(arr, start, end);
            if (pos == k) {
                return;
            } else if (pos > k) {
                findK(arr, start, pos, k);
            } else {
                findK(arr, pos, end, k);
            }
        }
    }
    /**
     * @param arr
     * @param i
     * @param len
     * @param k
     * @return
     */
    private int partion(int[] arr, int start, int end) {
        int left  = start; 
        int right  = end;
        int pivo = arr[left++];
        while(left < right){
            while(arr[right] >= pivo && left < right){
                right--;
            }

            while(arr[left] <= pivo && left < right){
                left++;
            }
            
            // 交换他们
            if(left < right){
                int tmp = arr[left];
                arr[left] = arr[right];
                arr[right] = tmp;
            }
        }
        // 最终 left 和 right 是会碰面的,这个位置就是哨兵要在的位置
        if(arr[left] < pivo){
            arr[start] = arr[left];
            arr[left]  = pivo;
        }
        return left;
    }
    /**
     * @param args
     */
    public static void main(String[] args) {
        int[] arr = new int[]{3,2,1};
        int k = 2;
//        int[] arr = new int[]{0,1,2,1};
//        int k = 1;
        new JianZhi40().getLeastNumbers(arr, k);
        for(int i = 0;i < k;i++){
            System.out.print(arr[i] +" ");
        }
    }

}

wangxiaoming CSDN认证博客专家 架构 Spring Boot Redis
博客是很好的总结和记录工具,如果有问题,来不及回复,关注微信公众号:程序员开发者社区,获取我的联系方式,向我提问,也可以给我发送邮件,联系 1275801617@qq.com
©️2020 CSDN 皮肤主题: Age of Ai 设计师: meimeiellie 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值