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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值