力扣 215. 数组中的第K个最大元素

🔗 leetcode.cn/problems/kt…

题目

• 返回数组 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;
    }
};