最小的k个数-CSDN博客

本文链接：https://blog.csdn.net/KobeSacre/article/details/127104512

这篇博客介绍了两种在Java中找出整数数组中最小k个数的方法。第一种是基于快速排序的思路，通过递归实现；第二种是利用优先队列（PriorityQueue），将数组元素依次加入队列并取出k个最小值。这两种解法分别展示了不同的算法思想，适用于不同场景。

摘要由CSDN通过智能技术生成

最小的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: 运用快排思维

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if (k == 0 || arr.length == 0) {
            return new int[0];
        }
        return quickSearch(arr, 0, arr.length - 1, k - 1);
    }

    private int[] quickSearch(int[] nums, int lo, int hi, int k) {
        int j = partition(nums, lo, hi);
        if (j == k) {
            return Arrays.copyOf(nums, j + 1);
        }
        return j > k? quickSearch(nums, lo, j - 1, k): quickSearch(nums, j + 1, hi, k);
    }
    
    private int partition(int[] nums, int left, int right) {
            int l = left, r = right;
            int cur = nums[left];
            while(l != r) {
                while(l < r && nums[r] >= cur) r--;
                while(l < r && nums[l] <= cur) l++;
                if(l < r) {
                    swap(nums, l, r);
                }            
            }
            swap(nums, left, r);
            return l;
    }

    
    private void swap(int[] nums, int l, int r) {
        int t = nums[l];
        nums[l] = nums[r];
        nums[r] = t;
    }
}

解法2: 优先队列

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if(arr.length == 0)
            return arr;
        
        PriorityQueue<Integer> queue = new PriorityQueue<>();
        
        for(int num : arr) {
            queue.offer(num);
        }
        
        int ans[] = new int[k];
        for(int i = 0; i < k;i++) {
            ans[i] = queue.poll();
        }
        
        return ans;
    }
}