[牛客题霸-研发] NC88-寻找第K大

----------------------------------------------本题链接----------------------------------------------

题目描述

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。

示例

输入
[1,3,5,2,2],5,3
 
返回值
2

思路

题目说的很清楚了，运用快速排序的思想，这里再给出快速排序代码：

该题可见[牛客题霸-研发] NC4-排序

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 将给定数组排序
     * @param arr int整型一维数组 待排序的数组
     * @return int整型一维数组
     */
    public int[] MySort (int[] arr) {
        if(arr.length == 0) return null;
        quickSort(arr, 0, arr.length-1);
        return arr;
    }
    
    private void quickSort(int[] arr, int left, int right){
        if(left >= right) return;
        int point = partition(arr, left, right);
        quickSort(arr, left, point-1);
        quickSort(arr, point+1, right);
    }
    
    private int partition(int[] arr, int left, int right){
        int first = arr[left];
        while(left < right){
            while(left < right && arr[right] >= first) right--;
            arr[left] = arr[right];
//             swap(arr, left, right);
            while(left < right && arr[left] <= first) left++;
            arr[right] = arr[left];
//             swap(arr, left, right);
        }
        arr[left] = first;
        return left;
    }
    
    private void swap(int[] arr, int i, int j){
        int tmp = arr[i];
        arr[i] = arr[j];
        arr[j] = tmp;
    }
}

算法过程

1. 判断异常情况
2. 快速排序找到中轴pivot等于K的位置
3. 输出结果

解答

public class Solution {
    public int findKth(int[] a, int n, int K) {
        return findK(a, 0, n-1, K);
    }
    
    private int partition(int[] arr, int left, int right){
        int pivot = arr[left];
        while(left < right){
            while(left < right && arr[right] <= pivot) right--;
            arr[left] = arr[right];
            while(left < right && arr[left] >= pivot) left++;
            arr[right] = arr[left];
        }
        arr[left] = pivot;
        return left;
    }
    
    private int findK(int[] arr, int left, int right, int k){
        if(left > right) return -1;
        
        int pivot = partition(arr, left, right);
        if(pivot == k-1) return arr[pivot];
        else if(pivot < k - 1) return findK(arr, pivot+1, right, k);
        else return findK(arr, left, pivot-1, k);
    }
}