【算法】【c++】【堆】347. Top K Frequent Elements

堆的典型例题
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;    
}

973. 最接近原点的 K 个点
     排序和堆两种方法