LeetCode 692 前K个高频单词[自定义排序 Map] HERODING的LeetCode之路

Posted 2021-05-23 HERODING23

解题思路：
统计字符串数目的问题首先想到就是map，但是难点在于怎么排序，因为map内进行排序的话需要构建新的map类，就略显复杂了，可不可以只定义一个排序函数就能够实现排序呢？当然可以！让vector承接map，对于sort来说，可以直接使用自定义的排序函数的，然后在vector中sort，输出前k项即可，代码如下：

bool compare(const pair<string, int>& s1, const pair<string, int>& s2) {
    // 频率相等，按字母排序
    if(s1.second == s2.second) return s1.first < s2.first;
    // 按照频率排序
    else return s1.second > s2.second;
}

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        unordered_map<string, int> count;
        // 统计频率
        for(string& word : words) {
            count[word] ++;
        }
        // 把map表存入vector中
        vector<pair<string, int>> res(count.begin(), count.end());
        // 按照自定义排序函数进行排序
        sort(res.begin(), res.end(), compare);
        vector<string> ans;
        // 保存最终结果
        for(int i = 0; i < k; i ++) {
            ans.push_back(res[i].first);
        } 
        return ans;
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/top-k-frequent-words/solution/c-mapzi-ding-yi-pai-xu-han-shu-by-herodi-7xfe/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/