2021-06-03Leetcode347.前K个高频元素

1.1我的想法
先用unordered_map统计每个数字出现的频率，以key作为数字，以value作为频率，对于value（频率）进行排序。最后将value相对应的数字，也就是key放入最终达到答案里面。
这里涉及到两个问题
1）是对于value进行排序，而map本身会对key进行排序，用sort也能对map的value进行排序
2）map不提供从value到key的查找

1.2 官方解答

class Solution {
public:
    static bool cmp(pair<int, int>& m, pair<int, int>& n) {
        return m.second > n.second;
    }
    //不明白为什么返回值要加上static
    
    struct cmp{
    bool operator()(pair<int,int>& m,pair<int,int>& n){
    	return m.second > n.second;
    }
    };

    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> occurrences;
        for (auto& v : nums) {
        //遍历容器的时候这里面加上引用是什么含义
        //for(auto &a:b)，循环体中修改a，b中对应内容也会修改
        //for(auto a:b)，循环体中修改a，b中内容不受影响
//for(const auto &a:b)，a不可修改，用于只读取b中内容
            occurrences[v]++;
        }
//先用哈希表统计出每一个元素的出现的频率
        // pair 的第一个元素代表数组的值，第二个元素代表了该值出现的次数
        priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(&cmp)> q(cmp);
        for (auto& [num, count] : occurrences) {
            if (q.size() == k) {
                if (q.top().second < count) {
                    q.pop();
                    q.emplace(num, count);
                }
            } else {
                q.emplace(num, count);
            }
        }
        vector<int> ret;
        while (!q.empty()) {
            ret.emplace_back(q.top().first);
            q.pop();
        }
        return ret;
    }
};

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/top-k-frequent-elements/solution/qian-k-ge-gao-pin-yuan-su-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

我的最终的解法
在设定优先队列的优先级的时候使用了functor.

class Solution{
    public:
    struct cmp{
        bool operator()(pair<int,int>& m,pair<int,int>& n){
            return m.second > n.second;
        }
    };
    //class 里面还可以包含class吗？
    //可以此时外界的class对于里面的class相当于一个namespace.
    vector<int> topKFrequent(vector<int>& nums,int k){
        unordered_map<int,int> occurrences;
        for(auto& i:nums){
            occurrences[i]++;
        }
        priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> q;
        for(auto&[num,count] : occurrences){
            if(q.size() == k){
                if(q.top().second < count){
                    q.pop();
                    q.emplace(num,count);
                }
            }
            else q.emplace(num,count);
        }
        vector<int> ret;
        while(!q.empty()){
            ret.push_back(q.top().first);
            q.pop();
        }
        return ret;
    }
};

最终学会的写法，较为简洁

class Solution {
public:
    struct cmp{
        bool operator()(pair<int,int>& m,pair<int,int>& n){
            return m.second > n.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> myMap;
        for(auto& i:nums){
            myMap[i]++;
        }
        priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> que;
        for(auto& p:myMap){
            que.push(p);
            if(que.size() > k)
            que.pop();
        }
        vector<int> ret;
        while(!que.empty()){
            ret.push_back(que.top().first);
            que.pop();
        }
        return ret;

    }
};

1.4学到了什么
1）map容器第三个参数可放置一个比较函数，根据比较函数的规则对key进行排序

template < class Key, class T, class Compare = less<Key>,
           class Allocator = allocator<pair<const Key,T> > > class map;

2）priority_queue优先队列
priority_queue，其中Type 为数据类型，Container为保存数据的容器，Functional 为元素比较方式。
Container必须是用数组实现的容器，比如vector,deque等等，但不能用 list。STL里面默认用的是vector。
对于优先队列的优先级的设定不是很明白