在未排序的数组中找到第 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 ≤ 数组的长度。
思路:
这道题我记得有一年的蓝桥杯代码填空题里出现过·,方法有很多种,一种是先排序直接取,还有一种是创建大根堆,一个个取,比较经典应该是利用类似快排的方法取,可以说是快排思想的一个应用,下面讲一讲这种方法。
利用类似快排的思想:设置一个参考值(一般是数组的第一个)和指向左右两边的双指针。找到右指针指向的值小于参考值的位置和左指针指向的值大于参考值的位置,交换它们,直到两指针相遇退出循环。然后在交换参考值和相遇位置的值。这样子就可以将数组以参考值分成两边,左边的值小于参考值,右边的值大于参考值。此时左右两边只是相对有序,只有参考值的位置是完全确定的,然后在判断k值与参考值位置的大于,看看要求的值在那一边,继续递归。
参考代码:
class Solution {
public int findKthLargest(int[] nums, int k)
{
int l=0,r=nums.length-1;
return getK(nums,nums.length-k,l,r);
}
private int getK(int[] nums, int k, int l, int r)
{
int ll=l,rr=r;
int n=nums[ll];
while(ll<rr)
{
while(ll<rr&&nums[rr]>=n)
rr--;
while(ll<rr&&nums[ll]<=n)
ll++;
swap(nums,ll,rr);
}
swap(nums, l, ll);
if(ll==k)
return n;
else if(ll>k)
return getK(nums, k, l, ll-1);
else
return getK(nums, k, ll+1, r);
}
private void swap(int[] nums,int l, int r)
{
int temp=nums[l];
nums[l]=nums[r];
nums[r]=temp;
}
}