C/C++教程

leetcode 692. 前K个高频单词 (百度)2021-08-03

本文主要是介绍leetcode 692. 前K个高频单词 (百度)2021-08-03,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

https://leetcode-cn.com/problems/top-k-frequent-words/

class Solution {
public:
    class cmp
    {
    public:
        bool operator()(const pair<string, int>& lhs, const pair<string, int>& rhs)
        {
            return lhs.second == rhs.second ? lhs.first < rhs.first : lhs.second > rhs.second;//注意这个地方,出现频率相同按字母排序
            // return lhs.second > rhs.second;
        }
    };

    vector<string> topKFrequent(vector<string>& words, int k) 
    {
        unordered_map<string, int> mp;
        for (auto& x: words)
        {
            mp[x]++;
        }
        //小顶堆 priority_queue
        priority_queue<pair<string, int>, vector<pair<string, int>>, cmp> prq;
        for (auto& y : mp)
        {
            // prq.emplace(y);
            prq.push(y);
            if (prq.size() > k)
                prq.pop();
        }

        vector<string> res(k);
        for (int i = k - 1; i >= 0; i--)
        {
            res[i] = prq.top().first;
            prq.pop();
        }
        return res;
    }
};
这篇关于leetcode 692. 前K个高频单词 (百度)2021-08-03的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!