Leetcode-692： 前K个高频单词

题目链接

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

题目

给一非空的单词列表，返回前 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 次。

注意

1. 假定 k 总为有效值， 1 ≤ k ≤ 集合元素数。
2. 输入的单词均由小写字母组成。

思路

用哈希表储存所有单词及其出现的次数，手撸个排序规则对其进行排序然后输出即可。

C++ Code

class Solution {
public:
    static bool cmp(pair<string,int> a, pair<string,int> b){
    if (a.second==b.second){
        return a.first<b.first;
    }
    else return a.second>b.second;
    }
    vector<string> topKFrequent(vector<string>& words, int k) {
        unordered_map<string, int> S;
        for(string s:words) S[s]++;
        sort(S.begin(),S.end(),cmp);
        vector<string> result;
        auto it = S.begin(); 
        while (k>=0)
        {
            string a =it->first;
            result.push_back(a);
            k--;it++;
        }
        return result;
    }

};

结果

报错，不能使用sort()对unordered_map结构进行排序。具体可参考：C++ 踩坑记录：不能使用sort()函数对unordered_map哈希表进行排序

 unordered_map结构无法使用sort()排序，但vector可以，因此，我们可以再构造一个vector，对unordered_map中的键排序。

C++ Code

class Solution {
public:
    static bool cmp(pair<string,int> a, pair<string,int> b){
    if (a.second==b.second){
        return a.first<b.first;
    }
    else return a.second>b.second;
    }
    vector<string> topKFrequent(vector<string>& words, int k) {
        unordered_map<string, int> S;
        for(string s:words) S[s]++;
        vector< pair<string, int> > V (S.begin(), S.end());
        sort(V.begin(),V.end(),cmp);
        vector<string> result;
        auto it = V.begin(); 
        while (k>0)
        {
            string a =it->first;
            result.push_back(a);
            k--;it++;
        }
        return result;
    }

};

结果