通过寻找第K大的数,分享一个快速排序的模板

题目:NC88 寻找第K大 ，来源牛客网

结合快排的图解，食用更佳。

题目描述

有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。 

给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间)，请返回第K大的数，保证答案存在。 

示例1

输入

[1,3,5,2,2],5,3

返回值

2

package nowcoder.研发最爱考;

/**
 * @Created by menghanjun
 * @Date 2021/2/26 11:32 上午
 */
public class NC88_寻找第K大 {
    public int findKth(int[] arr, int length, int K) {
        // write code here
        /**
         * 数组中查找，模板通常为排序
         * 此题使用快排来查找,跟快排不一样的地方就是可以丢弃掉k不在的那一半
         */
        return findK(arr, 0, length - 1, K);
    }

    private int findK(int[] arr, int left, int right, int target) {
        if (left > right) {
            return -1;
        }
        int pivot = partition(arr, left, right);
        int n = arr.length - pivot; //arr[pivot]代表第n大的数,比如0 1 2 3 4，pivot=4，那么arr[5]就代表第5-4大的数
        if (target < n) { //比如target=1，那就是找第一大的数，应该往右找
            return findK(arr, pivot + 1, right, target);
        } else if (target > n) {
            return findK(arr, left, pivot - 1, target);
        } else {
            return arr[pivot];
        }
    }

    /**
     * 快排的分区函数，将arr分为以pivot坐标为分界线的数组
     * pivot左侧都比arr[pivot]小
     * pivot右侧都比arr[pivot]大
     *
     * @param arr   数组
     * @param left  左边界
     * @param right 右边界
     * @return pivot的坐标
     */
    private int partition(int[] arr, int left, int right) {
        int i, j, pivot;
        i = left; //确定左指针
        j = right; //确定右指针
        pivot = left; //以左指针为基准
        while (i < j) { //这个循环的目的是保证左右指针重合的时候，重合位置左都是小的数，右都是大的数
            while (i < j && arr[j] >= arr[pivot]) { //一直找比基准小的数，右指针往左移动，找到跳出
                j--;
            }
            while (i < j && arr[i] <= arr[pivot]) { //一直找比基准大的数，左指针往右移动，找到跳出
                i++;
            }
            if (i < j) { //仍然合法
                swap(arr, i, j);
            }
        }
        swap(arr, pivot, i); //重合后将基准和重合位置交换
        return i; //返回新基准位置
    }

    private void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

    private void quickSort(int[] arr, int left, int right) {
        if (left > right) {
            return;
        }
        int pivot = partition(arr, left, right);
        quickSort(arr, left, pivot - 1);
        quickSort(arr, pivot + 1, right);
    }
}