一、题目描述:215. 数组中的第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、快排的partion思想,复杂度nlog(n)
2、最小堆,复杂度nlog(k);
三、代码(方案一)
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
return partion(nums,0,nums.size()-1,k);
}
int partion(vector<int>& nums,int left,int right,int k)
{
int tmp = nums[left];
int l = left;
int r =right;
while(l<r)
{
while(l<r&&nums[r]<= tmp)
{
r--;
}
if(l<r && nums[r] > tmp)
{
swap(nums[l],nums[r]);
//r--;
}
while(l<r&&nums[l]>tmp)
{
//nums[r] = t;
l++;
}
if(l<r && nums[l] <= tmp)
{
swap(nums[l],nums[r]);
//r--;
}
}
//print(nums,l);
nums[l] = tmp;
if(l == k-1)
return nums[l];
if(l < k-1)
{
return partion(nums,l+1,right,k);
}
else
{
return partion(nums,left,l-1,k);
}
}
void print(vector<int>&nums,int k)
{
cout<<"k="<<k<<endl;
for(int i =0 ;i<nums.size();i++)
{
cout<<nums[i]<<",";
}
cout<<endl;
}
void swap(int & left,int & right)
{
int tmp = left;
left = right;
right = tmp;
}
};