Top K Frequent Words（C++前K个高频单词）

最新推荐文章于 2021-05-29 16:16:42 发布

coolsunxu

最新推荐文章于 2021-05-29 16:16:42 发布

阅读量215

点赞数

分类专栏： C++ LeetCode 文章标签： leetcode 队列 C++ map

本文链接：https://blog.csdn.net/coolsunxu/article/details/105450156

版权

C++ 同时被 2 个专栏收录

735 篇文章 18 订阅

订阅专栏

LeetCode

459 篇文章 1 订阅

订阅专栏

参考网址：https://leetcode.com/problems/top-k-frequent-words/discuss/108366/O(nlog(k))-Priority-Queue-C%2B%2B-code

解题思路：

（1）使用unordered_map的键值对来存储字符串和对应出现的次数

（2）再重写priority_queue（优先队列）的比较函数，将优先级较大的放在队首，依次取前k个即可



class Solution {
public:
    
    vector<string> topKFrequent(vector<string>& words, int k) {
        unordered_map<string, int> freq;
        for(auto w : words){
            freq[w]++;
        }
    
        auto comp = [&](const pair<string,int>& a, const pair<string,int>& b) {
    	    //return true if b is considered to go top a
            return a.second < b.second || (a.second == b.second && a.first > b.first); 
        };
        priority_queue< pair<string,int>, vector<pair<string,int>>, decltype(comp) > pq(freq.begin(),freq.end(),comp);
    
    
        vector<string> output;
    
        while(k!=0){
            output.push_back(pq.top().first);
            pq.pop();
            k--;
        }
        return output;
    }
};