前K个高频单词

给一非空的单词列表,返回前 k 个出现次数最多的单词。

返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。

示例 1:

输入: ["i", "love", "leetcode", "i", "love", "coding"], k = 2
输出: ["i", "love"]
解析: "i" 和 "love" 为出现次数最多的两个单词,均为2次。
    注意,按字母顺序 "i" 在 "love" 之前。
 

示例 2:

输入: ["the", "day", "is", "sunny", "the", "the", "the", "sunny", "is", "is"], k = 4
输出: ["the", "is", "sunny", "day"]
解析: "the", "is", "sunny" 和 "day" 是出现次数最多的四个单词,
    出现次数依次为 4, 3, 2 和 1 次。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-words
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

vector<string> topKFrequent(vector<string>& words, int k) {
    map<string,int> mp;
    vector<string> res;
    for (auto& item : words)
    {
        mp[item]++;
    }
    //由于map不支持value的sort排序,所以必须将map转换为vector才能进行sort自定义排序
    vector<pair<string, int>> vec(mp.begin(), mp.end());
    //sort的第三个参数是lambda表达式,其实和传普通函数一样,sort的第三个参数是支持二元谓词的
    sort(vec.begin(), vec.end(), []
        (const pair<string,int> &left, const pair<string,int> &right)
        {
            if (left.second == right.second)
                return left.first < right.first;
            else
                return left.second > right.second;
        });
    //将vector的前k个元素添加到res中
    for (const auto &item : vec)
    {
        if (k-- == 0)
            break;
        res.push_back(item.first);
    }
    return res;
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值