215.数组中的第k个最大元素
解题思路:
1.基于快排的的选择
(快排时间复杂度是O(n*logn))
但我们实际上只需要找到倒数第k个元素,快排是基于partition的分治法,我们只需要找到某次划分得到的index正好是我们需要的k即可
2.基于堆排序的选择方法
package leadcode;
/**
* @author : icehill
* @description : 数组中的第K个最大元素
* 在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
* 示例 1:
* 输入: [3,2,1,5,6,4] 和 k = 2
* 输出: 5
* 示例 2:
* 输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
* 输出: 4
* 说明:
* 你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度
* 来源:力扣(LeetCode)
* 链接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array
* 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
* 解题思路:
* 1.基于快排的选择法
* 时间复杂度:O(n) 空间复杂度O(logn)
* @date : 2021-05-08
*/
public class Solution215 {
public static void main(String[] args) {
Solution215 solution215 = new Solution215();
int[] nums = {3, 2, 1, 5, 6, 4};
// int[] nums = {3,2,3,1,2,4,5,5,6};
System.out.println(solution215.findKthLargest(nums, 2));
System.out.println(solution215.findKthLargest(nums, 4));
}
public int findKthLargest(int[] nums, int k) {
int destIndex = nums.length - k;
int l = 0;
// 转换一下,第 k 大元素的索引是 len - k
int r = nums.length - 1;
while (true) {
int index = partition(nums, l, r);
if (index == destIndex) {
return nums[index];
} else if (index > destIndex) {
r = index - 1;
} else {
l = index + 1;
}
}
}
/**
* 在数组 nums 的子区间 [left, right] 执行 partition 操作,返回 nums[left] 排序以后应该在的位置
* 在遍历过程中保持循环不变量的语义
* 1、[left + 1, j] < nums[left]
* 2、(j, i] >= nums[left]
*
* @param nums
* @param l
* @param r
* @return
*/
public int partition(int[] nums, int l, int r) {
int pivot = nums[l];
int j = l;
for (int i = l + 1; i <= r; i++) {
if (nums[i] < pivot) {
j++;
//比基准元素小的都放到前面,然后指针j++,[l,j]都是比pivot小的元素
swap(nums, j, i);
}
}
swap(nums, j, l);
return j;
}
/**
* 交换两个元素
*
* @param nums
* @param a
* @param b
*/
public void swap(int[] nums, int a, int b) {
int temp = nums[a];
nums[a] = nums[b];
nums[b] = temp;
}
}