题目来源:力扣
题目描述:
给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
代码:
int findKthLargest(vector<int>& nums, int k) {//Npriority_queue<int> pq(nums.begin(),nums.end());//K*logNwhile(--k){pq.pop();}return pq.top();}
这道题我们可以建大堆,也可以建小堆,这是大堆的玩法
当数据量较大时,尤其是文件这种,是需要建小堆的,下面我们来建小堆
int findKthLargest(vector<int>& nums, int k) {//Npriority_queue<int,vector<int>,greater<int>> pq(nums.begin(),nums.begin()+k);//(N-K)*logKfor(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是一个函数模板,要传对象,我们传的是匿名对象,而下面的是类模板,要传类型,所以不加括号