题目:
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
法一:排序后输出。
法二:优先队列(堆)。
法三:利用快速排序的方法。
在这里仅介绍法三,法一法二调用库函数即可。
思路:
首先,找第k个最大的元素,即是找第n-k+1个最小的元素。
步骤一:
我们利用快排的思想,每一次对一个元素进行归位,判断该元素最终位置p与n-k+1位置关系。
步骤二:
1、n-k+1>p:则对p右侧元素进行步骤一操作。
2、n-k+1<p:则对p左侧元素进行步骤一操作。
直到n-k+1 = p停止。
public int findKthLargest(int[] nums, int k) {
if(nums.length==1) {
return nums[0];
}
helper(nums,0,nums.length-1,nums.length-k);
return nums[nums.length-k];
}
public void helper(int[] nums,int lo,int hi,int k){
int i = lo;
int j = hi+1;
int v = nums[lo];
while(true) {
while(i+1<nums.length&&nums[++i]<v) {if(i==hi)break;}
while(j-1>=0&&nums[--j]>v);
if(i>=j) {break;}
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
int temp = nums[j];
nums[j] = v;
nums[lo] = temp;
if(j>k) {
helper(nums,lo,j-1,k);
}
else if(j<k) {
helper(nums,j+1,hi,k);
}
}