堆的典型例题
Given a non-empty array of integers, return the k most frequent elements.
class Solution {
public:
static bool comp(pair<int, int> a, pair<int, int> b){
return a.second > b.second; //小顶堆要大于号
}
vector<int> topKFrequent(vector<int>& nums, int k) {
//要选出前k大的元素 就是建立一个小顶堆
//在遍历的过程中剩下的就是要求的
//先把元素求出来
unordered_map<int, int> mp;
for(auto n: nums) {
mp[n]++;
}
priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(&comp)> q(comp);
for(auto& p: mp){
if(q.size() >= k){
//判断是否替换堆顶
if(p.second > q.top().second) {
q.pop();
q.push(p);
}
}else{
//直接放入
q.push(p);
}
}
vector<int> res;
while(!q.empty()){
res.push_back(q.top().first);
q.pop(); //queue不能随机存取
}
return res;
}
};
简简单单试验一下堆
#include<bits/stdc++.h>
using namespace std;
bool cmp(int& a, int& b) {
if(a < b) {
return true;
}
return false;
}
int main() {
priority_queue<int, vector<int>, decltype(&cmp)> q(cmp);
for(int i = 0; i < 5; ++i) {
q.push(20 - i);
}
int temp;
while (!q.empty())
{
temp = q.top();
q.pop();
cout << temp << endl;
}
return 0;
}
- 最接近原点的 K 个点
排序和堆两种方法