难度等级:中等
上一篇算法:
88. 合并两个有序数组【184】
力扣此题地址:
215. 数组中的第K个最大元素 - 力扣(Leetcode)
给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。


此题基于快速排序进行改编,来求解目标元素。
快速排序的原理是找一个基准值,然后把小于基准值的所有元素放在左边,大于基准值的所有元素放在右边,然后再用递归的方式分别对左右两边的元素进行快速排序。
在此题中,因为找的是第k个最大的元素,所以目标值的索引位置为array.length-k,所以每排一次序之后,判断基准值的索引位置是否小于目标值的索引位置,小于的话则对右边的元素进行排序,否则对左边的元素进行排序,依次类推,当基准值的索引位置等于目标值的索引位置时,将其位置对应的值返回即可。
这样的话,原本的快速排序需要对所有的元素都进行排序,改编之后每次排序只需要对基准值其中一边的元素进行排序即可。
其时间复杂度为O(logn)
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)