215.数组中的第K个最大元素-C++

KLZUQ2023-07-29 8:47

题目来源：力扣

题目描述：

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

代码：

cpp 复制代码

    int findKthLargest(vector<int>& nums, int k) {
        //N
        priority_queue<int> pq(nums.begin(),nums.end());
        //K*logN
        while(--k){
            pq.pop();
        }
        return pq.top();
    }

这道题我们可以建大堆，也可以建小堆，这是大堆的玩法

当数据量较大时，尤其是文件这种，是需要建小堆的，下面我们来建小堆

cpp 复制代码

    int findKthLargest(vector<int>& nums, int k) {
        //N
        priority_queue<int,vector<int>,greater<int>> pq(nums.begin(),nums.begin()+k);
        //(N-K)*logK
        for(int i=k;i<nums.size();i++){
            if(nums[i]>pq.top()){
                pq.pop();
                pq.push(nums[i]);
            }
        }
        return pq.top();
    }

这是小堆的玩法，我们对前k-1个数据建堆，然后从第k个数据开始，如果这个数据比堆顶数据要大，就替换这个数据，这样我们得到的就是由整个数组中前k大的数据构建成的小堆，而堆顶自然是第k大的数据

这道题我们其实还可以直接排序过的，大家遇到实在不会的，可以先找个简单的方法试一试

这里有一个问题

我们发现，sort里传的有括号，而下面的优先级队列里没有括号

原因是上面的sort是一个函数模板，要传对象，我们传的是匿名对象，而下面的是类模板，要传类型，所以不加括号