第k大元素-快速排序

第k大元素-快速排序

在数组中找到第 k 大的元素。

样例 1：

输入：
n = 1, nums = [1,3,4,2]
输出：
4

样例 2：

输入：
n = 3, nums = [9,3,2,4,8]
输出：
4

挑战
要求时间复杂度为O(n)，空间复杂度为O(1)。

注意事项
你可以交换数组中的元素的位置

分析

这里使用快速排序

代码

public class Solution {
    /**
     * @param n: An integer
     * @param nums: An array
     * @return: the Kth largest element
     */
    public int kthLargestElement(int n, int[] nums) {
        return sortA(nums)[nums.length-n];
    }
    public int[] sortA(int[] sourceArray) {
        int[] arr = Arrays.copyOf(sourceArray, sourceArray.length);
        return quickSort(arr, 0, arr.length - 1);
    }
    private int[] quickSort(int[] arr, int left, int right) {
        if (left < right) {
            int partitionIndex = partition(arr, left, right);
            quickSort(arr, left, partitionIndex - 1);
            quickSort(arr, partitionIndex + 1, right);
        }
        return arr;
    }

    private int partition(int[] arr, int left, int right) {
        int pivot = left;
        int index = pivot + 1;
        for (int i = index; i <= right; i++) {
            if (arr[i] < arr[pivot]) {
                swap(arr, i, index);
                index++;
            }
        }
        swap(arr, pivot, index - 1);
        return index - 1;
    }

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

快速排序

从数列中挑出一个元素，称为 “基准”（pivot）;

重新排序数列，所有元素比基准值小的摆放在基准前面，所有元素比基准值大的摆在基准的后面（相同的数可以到任一边）。在这个分区退出之后，该基准就处于数列的中间位置。这个称为分区（partition）操作；

递归地（recursive）把小于基准值元素的子数列和大于基准值元素的子数列排序；

关于第二步的具体代码步骤

取第一个数作为基准数
每个数和第一个数对比，大的不变，小的和标志位交换位置，这时候标志位加+1，这样如果有比基准数大的数，就会被放到标志位的地方（如果前面没有比它大的数，则和它自己交换位置，即不变）这时候标志位加+1。最后交换标志位-1和第一个数，标志位减一即为标志数，所对应的值为基准值。

还有另外一种写法，开始没有基准数
先取标志位为0，从一开始每个数和标志位对比，大则不变，小则这位=标志位+1，标志位+1=标志位，标志位=这位，这时候标志位加一。最后标志位所在的数就是基准数。
就是一个位移操作，把小于标志数插入标志位前面，然后把标志位后移。