在未排序的数组中找到第 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
首先找出第K个最大元素,可以直接把数组排序,再从数组索引最后一个向内移动,直到第K个元素即可输出。这样做的话复杂度就直接是O(nlgn)的复杂度
快速选择:
类似快速排序的操作将前K大的元素都放在数组最右边 定义nums[0]等于i(数组起始为head,末尾为tail) 利用快速排序的双指针,将他置于某一中间位置,用数组的长度减去这时的位置即可得到大于i 元素的个数,使其与k比较,若大于k则说明包含位置靠前了,大于i的元素包含k了,则从i的位置+1 到tail再进行一次选择 反之则位置靠后,则从head到i的位置-1再进行一次选择,当数组长度减k=0的时候则说明所有比k大的元素都放在了k的后面数组该位置的元素即可
public class Solution {
int nums[] ;
public int findKthLargest(int[] nums, int k) {
this.nums = nums ;
quickSelect(0,nums.length-1,k);
return nums[nums.length - k ];
}
//类似于快速排序的双指针
void quickSelect( int head, int tail, int k ) {
int i = this.nums[head];
int x = head ;
int y = tail ;
while (x != y) {
while (x < y && i <= this.nums[y]) {
y-- ;
}
swap(x,y);
while (y > x && i >= this.nums[x]) {
x++;
}
swap(x,y);
}
//对于指针位置的判断,因为x=y的所以判断一个就可以了
if (k == nums.length-x) {
return;
} else if (nums.length-x > k) {
quickSelect(x+1,tail,k);
} else {
quickSelect(head,x-1,k);
}
}
//交换
void swap( int i, int j) {
int temp = this.nums[i];
this.nums[i] = this.nums[j];
this.nums[j] = temp ;
}
}
复杂度分析:平均情况O(N),最坏情况 O(N2)
想是这样写,写出来真的慢的离谱。。。还不如直接调库排序然后输出第K个元素快。。。