LeetCode 面试题 17.14. 最小K个数 【快速排序】

题目

设计一个算法，找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例：

输入： arr = [1,3,5,7,2,4,6,8], k = 4
输出： [1,2,3,4]
提示：

0 <= len(arr) <= 100000
0 <= k <= min(100000, len(arr))

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/smallest-k-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

快速排序

class Solution {
    void quick_sort(int[] arr,int l,int r,int k){
        if(l>=r) return;
        int x=arr[(l+r)/2],i=l-1,j=r+1;
        while(i<j){
            do i++;while(arr[i]<x);
            do j--;while(arr[j]>x);
            if(i<j){
                int t=arr[i];
                arr[i]=arr[j];
                arr[j]=t;
            }
        }
        int sl=j-l+1;
        quick_sort(arr,l,j,k);
        if(k>sl) quick_sort(arr,j+1,r,k-sl); 
    }
    public int[] smallestK(int[] arr, int k) {
        int[] ans = new int[k];
        if(k==0) return ans;
        quick_sort(arr,0,arr.length-1,k);
        for(int i=0;i<k;i++){
            ans[i]=arr[i];
        }
        return ans;
    }
}