215. 数组中的第K个最大元素【382】

难度等级：中等

上一篇算法：

88. 合并两个有序数组【184】

力扣此题地址：

215. 数组中的第K个最大元素 - 力扣（Leetcode）

1.题目:215. 数组中的第K个最大元素

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

2.解题思路：

此题基于快速排序进行改编，来求解目标元素。

快速排序的原理是找一个基准值，然后把小于基准值的所有元素放在左边，大于基准值的所有元素放在右边，然后再用递归的方式分别对左右两边的元素进行快速排序。

在此题中，因为找的是第k个最大的元素，所以目标值的索引位置为array.length-k，所以每排一次序之后，判断基准值的索引位置是否小于目标值的索引位置，小于的话则对右边的元素进行排序，否则对左边的元素进行排序，依次类推，当基准值的索引位置等于目标值的索引位置时，将其位置对应的值返回即可。

这样的话，原本的快速排序需要对所有的元素都进行排序，改编之后每次排序只需要对基准值其中一边的元素进行排序即可。

其时间复杂度为O（logn）

3.代码实现：

class Solution {
    Random random = new Random();

    public int findKthLargest(int[] nums, int k) {
        return quickSelect(nums, 0, nums.length - 1, nums.length - k);
    }

    public int quickSelect(int[] a, int l, int r, int index) {
        int q = randomPartition(a, l, r);
        if (q == index) {
            return a[q];
        } else {
            return q < index ? quickSelect(a, q + 1, r, index) : quickSelect(a, l, q - 1, index);
        }
    }

    public int randomPartition(int[] a, int l, int r) {
        int i = random.nextInt(r - l + 1) + l;
        swap(a, i, r);
        return partition(a, l, r);
    }

    public int partition(int[] a, int l, int r) {
        int x = a[r], i = l - 1;
        for (int j = l; j < r; ++j) {
            if (a[j] <= x) {
                swap(a, ++i, j);
            }
        }
        swap(a, i + 1, r);
        return i + 1;
    }

    public void swap(int[] a, int i, int j) {
        int temp = a[i];
        a[i] = a[j];
        a[j] = temp;
    }
}

思路参考：215. 数组中的第K个最大元素 - 力扣（Leetcode）