力扣 692.前K个高频单词

高自期许

于 2023-01-20 15:26:39 发布

阅读量107

点赞数

分类专栏：模拟文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_44014982/article/details/128741948

版权

模拟专栏收录该内容

32 篇文章 1 订阅

订阅专栏

该文章介绍了一个LeetCode问题，涉及如何使用C++中的vector和map数据结构，以及自定义排序函数cmp，来找出给定单词列表中出现频率最高的k个单词。首先，利用unordered_map存储每个单词的频率，然后将map中的元素转换为pair类型的vector并排序，最后返回前k个高频词。

摘要由CSDN通过智能技术生成

题目链接：https://leetcode.cn/problems/top-k-frequent-words/
在这里插入图片描述
借助vector对map进行排序

int cmp(pair<string,int>& a, pair<string,int>& b){
    if (a.second != b.second)return a.second > b.second;
    else return a.first < b.first;
}

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        vector<string> vec;
        vector<pair<string, int> > arr;
        unordered_map<string, int> m;
        for (int i = 0; i < words.size(); ++i) {
            ++m[words[i]];
        }
        for (auto i = m.begin(); i != m.end(); ++i) {
            arr.push_back(*i);
        }
        sort(arr.begin(), arr.end(), cmp);
        for (int i = 0; i < arr.size(); ++i) {
            vec.push_back(arr[i].first);
            if (vec.size() == k)break;
        }
        return vec;
    }
};