最小的k个数

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

1.常规解法

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        Arrays.sort(arr);
        int len=arr.length;
        int[] a=new int[k];
        int t=0;
        for(int i=0;i<k;i++)
        {
            a[t++]=arr[i];

        }
        return a;

    }
}

 

 

2.快速排序法

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        quickSort(arr,0,arr.length-1);
        return Arrays.copyOf(arr,k);
       
    }

    public void quickSort(int[] arr,int low,int high){
        if(low>=high)return;
        int pivotPos=partition(arr,low,high);
        quickSort(arr,low,pivotPos-1);
        quickSort(arr,pivotPos+1,high);



    }

    int partition(int[] arr,int low,int high){
        int pivot=arr[low];
        while(low<high)
        {
            while(low<high&&arr[high]>=pivot)high--;
            arr[low]=arr[high];
            while(low<high&&arr[low]<=pivot)low++;
            arr[high]=arr[low];


        }
        arr[low]=pivot;
        return low;



    }
}

 

上一篇:快速排序


下一篇:基本排序算法——快速排序