力扣 215. 数组中的第K个最大元素
🔗 https://leetcode.cn/problems/kth-largest-element-in-an-array
题目
- 返回数组 nums 中的第 k 大数
思路
- 桶排,把数据都调整为正数,放置到对应的桶位置,最后遍历桶获得第 K 大的数
代码
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
int mark[20010];
memset(mark, 0, sizeof mark);
int offset = 10000;
for (auto num : nums) {
num += offset;
mark[num]++;
//printf("num %d true\n", num);
}
k = nums.size() - k + 1;
int count = 0;
int ans = 0;
for (int i = 0; i< 20010; i++) {
count += mark[i];
if (count >= k) {
ans = i;
break;
}
}
return ans - offset;
}
};