LeetCode；数组中第K大的元素；标签：排序

MOXI33441

于 2023-01-11 20:00:09 发布

阅读量52

点赞数

分类专栏：排序文章标签：算法 Powered by 金山文档

本文链接：https://blog.csdn.net/MOXI33441/article/details/128650082

版权

排序专栏收录该内容

5 篇文章 0 订阅

订阅专栏

该代码实现了一个算法，通过哈希表在O(n)时间内找到数组中第k个最大的元素，避免了排序导致的时间复杂度增加。首先统计每个数值的出现次数，然后从大于0的数值开始检查，更新k值，直到找到第k大的数。如果大于0的部分未找到，再检查小于0的部分。

摘要由CSDN通过智能技术生成

这里没用排序做，用的哈希表，涉及到排序的话时间复杂度都大于o(n)了，不满足条件了

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        // 这个题不能用到排序，只要用到排序时间复杂度必然大于o(n);
        vector<int> count(20001,0);
        // 我们记录每个值出现的次数
        for(auto it = nums.begin();it!=nums.end();it++){
            if(*it>=0){
                count[*it]++;
            }else{
                count[-*it+10000]++;
            }
        }
        // 先遍历大于0的部分
        for(int i=10000;i>=0;i--){
            // 值i存在
           if(count[i]>0){
               k-=count[i];
               if(k<=0){
                   // 这就是第k大的数
                   return i;
               }
           }
        }
        // 如果大于零的没找到，遍历小于0的部分
        for(int i=10001;i<count.size();i++){
            if(count[i]>0){
               k-=count[i];
               if(k<=0){
                   return 10000-i;
               }
            }
        }
        return 0;
    }
};