Leetcode 692. 前K个高频单词

JehanRio

已于 2023-10-07 14:49:27 修改

阅读量329

点赞数 1

分类专栏： Leetcode专栏文章标签： leetcode 算法职场和发展

于 2023-09-10 09:49:23 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/132788059

版权

Leetcode专栏专栏收录该内容

278 篇文章 0 订阅

订阅专栏

文章目录

题目
代码（9.10 首刷自解）
代码（10.7 二刷自解）

题目

在这里插入图片描述
Leetcode 692. 前K个高频单词

代码（9.10 首刷自解）

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        unordered_map<string, int> mp;
        for(string& word : words) {
            mp[word]++;
        }
        auto cmp = [](auto a, auto b) {
            if(a.second == b.second)
                return a.first > b.first;
            return a.second < b.second;
        };
        priority_queue<pair<string, int>, vector<pair<string, int>>, decltype(cmp)> q(cmp);
        vector<string> ans;
        for(auto& kv : mp) {
            q.emplace(kv.first, kv.second);
        }
        while(k--) {
            ans.emplace_back(q.top().first);
            q.pop();
        }
        return ans;
    }
};

代码（10.7 二刷自解）

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        unordered_map<string, int> mp;
        for(string& s : words)
            mp[s]++;
        auto cmp = [](auto& a, auto& b) {
            if(a.second == b.second)
                return a.first > b.first;
            return a.second < b.second;
        };
        priority_queue<pair<string, int>, vector<pair<string, int>>, decltype(cmp)> q(cmp);
        for(auto kv : mp)
            q.push(kv);
        vector<string> res;
        for(int i = 0; i < k; i++) {
            res.emplace_back(q.top().first);
            q.pop();
        }
        return res;
    }
};