在未排序的数组中找到第 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 ≤ 数组的长度。
解法一:冒泡排序,从后往前排,复杂度O(kn)
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
int n = nums.size();
if(n==0){
return 0;
}
int start = 0;
while(start<k){
for(int i=n-1;i>start;--i){
if(nums[i]>nums[i-1]){
int temp = nums[i-1];
nums[i-1] = nums[i];
nums[i] = temp;
}
}
start++;
}
return nums[k-1];
}
};
解法二:小顶堆,复杂度O(nlogk)
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
int N=nums.size();
build_Ksmallstack(nums,k);
for(int index=k;index<N;++index){
if(nums[index]>nums[0]){
swap(nums,index,0);
smallstack(nums, k, 0);
}
}
return nums[0];
}
void build_Ksmallstack(vector<int>& nums, int k){
int noTreeNodes = (k-1)/2;
for(int i=noTreeNodes;i>=0;--i){
smallstack(nums,k,i);
}
return;
}
void smallstack(vector<int>& nums, int k, int index){
int min = index;
while(true){
if(2*index+1<k && nums[2*index+1]<nums[index]){
min = 2*index+1;
}
if(2*index+2<k && nums[2*index+2]<nums[min]){
min = 2*index+2;
}
if(min==index){
break;
}
swap(nums,min,index);
index = min;
}
}
void swap(vector<int>& nums,int min,int index){
int temp = nums[min];
nums[min] = nums[index];
nums[index] = temp;
return;
}
};