一、题目简述
给定一个未排序的序列,找到其第K大的元素。
示例:
给定[3,2,1,5,6,4],k=2,返回5。
注意:
假定K总是有效的,即, 1≤k≤lens
函数原型:int findKthLargest(vector<int>& nums, int k)
二、编程思路
本题的编程思路类似归并排序,对于一个序列 nums ,可以随机选择其中一个元素 val ,为方便考虑,我们选取序列中心元素作为 val ,将大于 val 的数字放入 lt_vct(larger than vector) 中,等于的放入 eq_vct(equal vector) 中,小于的放入 st_vct(smaller than vector) 中。
则
vector长度 lt_vct_len eq_vct_len st_vct_eq 下标范围 [0,lt_vct_len−1] [lt_vct_len,lt_vct_len+eq_vct_len−1] [lt_vct_len+eq_vct_len,lt_vct_len+eq_vct_len+st_vct_eq−1]
- 若k落在eq_vct的范围内,说明 val 即是第K大元素,返回 val ;
- 若k落在lt_vct的范围内,说明第K大元素在lt_vct中,则在lt_vct中继续寻找;
- 若k落在st_vct的范围内,说明第K大元素是st_vct中的第K-lt_vct_len-eq_vct_len大元素;
注意:数组下标范围及与K的关系。
三、代码设计、
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
return kthLargest(nums, k);
}
int kthLargest(vector<int>& nums, int k){
int end = nums.size()-1;
vector<int> larger, smaller,equal;
int mid_idx = floor(end / 2);
//cout << nums[mid_idx] << endl;
for (int i = 0; i < nums.size(); i++){
if (nums[i] > nums[mid_idx]){
larger.push_back(nums[i]);
}
else if (nums[i] < nums[mid_idx]){
smaller.push_back(nums[i]);
}
else{
equal.push_back(nums[i]);
}
}
int lt_size, st_size, eq_size;
lt_size = larger.size();
st_size = smaller.size();
eq_size = equal.size();
if ((k-1 >= lt_size) && (k-1 <= eq_size + lt_size-1)){
//递归基
//说明nums[mid_idx]为第K大的数
return nums[mid_idx];
}
//递归步
else if((k-1<=lt_size-1) && (k-1>=0)){
return kthLargest(larger, k);
}
else if((k-1>=lt_size+eq_size)&& (k-1<=lt_size+eq_size+st_size-1)){
return kthLargest(smaller, k - lt_size - eq_size);
}
else{
cout << "bug here!" << endl;
return -9999;
}
}
};
四、编程心得
数组要小心下标,递归要有递归基。