215. Kth Largest Element in an Array

最新推荐文章于 2022-07-02 13:48:51 发布

原创最新推荐文章于 2022-07-02 13:48:51 发布 · 152 阅读

0 ·

CC 4.0 BY-SA版权

C++ 同时被 2 个专栏收录

45 篇文章

订阅专栏

leetcode

37 篇文章

订阅专栏

本文介绍了三种寻找数组中第K大元素的方法：使用哈希表排序、优先级队列以及调用C++自带函数。通过对比，发现优先级队列方法效率最高。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道题虽然难度不大，而且可以用不同方法解决，但是确实学到了不少东西，尤其是第一次看到了优先级队列这种操作。

第一种方法，hash表排序：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        if (nums.size()<1 || k>nums.size()) {
            return 0;
        }
        multiset<int> mset;
        
        for (int i=0; i<nums.size(); ++i) {
            mset.insert(nums[i]);
        }
        
        int index = 0;
        
        for (auto& ans : mset) {
            if (index == nums.size()-k) {
                return ans;
            }
            ++index;
        }
        return 0;
    }
};

第二种方法，优先级队列：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        if (nums.size()<1 || k>nums.size()) {
            return 0;
        }
        priority_queue<int, vector<int>, greater<int>> numspq;
        
        for (int i=0; i<nums.size(); ++i) {
            numspq.push(nums[i]);
            if (numspq.size() > k) {
                numspq.pop();
            }
        }
        return numspq.top();
    }
};

第三中方法，调用C++自带函数：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        if (nums.size()<1 || k>nums.size()) {
            return 0;
        }
        
        nth_element(nums.begin(), nums.begin()+(nums.size()-k), nums.end());
        
        return nums[nums.size()-k];
    }
};

而且自带的方法效率不底，最高的效率是优先级队列。