【OJ - map】前K个高频单词

题目难度：中等

OJ链接：692. 前K个高频单词 - 力扣（LeetCode） (leetcode-cn.com)

题目描述：

给定一个单词列表 words 和一个整数 k ，返回前 k 个出现次数最多的单词。

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

示例：

输入: words = [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2

输出: [“i”, “love”]

解析: “i” 和 “love” 为出现次数最多的两个单词，均为2次。（注意，按字母顺序 “i” 在 “love” 之前。）

---

解题思路1：

思路如图1、2、3：

这里出现的问题是：

• 调用 sort 对单词出现次数排降序，因为sort底层是快排，不稳定，所以对单词出现次数排好序后，单词的字典序却乱了。

如何解决：

• 想要保证稳定性（即对单词出现次数排序时，单词原先已经排好的字典序不会变），得用冒泡，直接插入，归并稳定的排序算法，可惜官方库并没有提供这些函数，不能直接用。
• 或者把每一组出现次数相同的单词集中起来放入map中，这些出现次数相同的单词就按照字典序排好序了

下面代码是不完整的，解决上述问题的部门还没有写：

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        // 1、把<key-单词, value-单词出现次数>键值对插入map中
        // 当插入完成后，单词已经按照字典序排序了
        map<string, int> count_map;
        for(auto& e : words)
        {
            count_map[e]++;
        }
        
        // 2、接下来用sort对单词出现次数排降序
        // 把map迭代器插入vector中
        vector<map<string, int>::iterator> kvArr;
        for(map<string, int>::iterator it = count_map.begin(); it != count_map.end(); it++)
        {
            kvArr.push_back(it);
        }

        // 对map迭代器指向元素中的value值（即单词出现次数）进行比较的仿函数
        struct IterCompare
        {
            bool operator()(map<string, int>::iterator it1, map<string, int>::iterator it2)
            {
                return it1->second > it2->second;
            }
        };
        sort(kvArr.begin(), kvArr.end(), IterCompare());

        //......
        // 这里把单词出现次数相同的找出来放入map中，此时出现次数相同的单词就按照字典序排好了
        //......
        
        
        // 取出vector中前k个map迭代器指向元素的键值key，放入到ret中，并返回结果
        vector<string> ret;
        for(size_t i = 0; i < kvArr.size() && i < k; i++)
        {
            ret.push_back(kvArr[i]->first);
        }
        return ret;
    }
};

---

解题思路2：最优解

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        // 把<单词, 单词出现次数>键值对插入map中
        map<string, int> count_map;
        for(auto& e : words)
        {
            count_map[e]++;
        }

        // 当插入完成后，map中的单词已经按照字典序排好序了
        /*
        <coding, 1>
        <i, 2>
        <leetcode, 1>
        <love, 2>
        */
        
        // 现在需要在不破坏单词原顺序的情况下，按照单词出现次数排降序
        /*
        <2, i>
        <2, love>
        <1, coding>
        <1, leetcode>
        */
        
        // 用multimap对单词出现次数排序，原有的单词顺序被不会打乱，相当于是稳定的
        multimap<int, string, greater<int>> sort_map;
        for(auto& e : count_map)
        {
            // 把<单词出现次数, 单词>键值对插入map中
            sort_map.insert(make_pair(e.second, e.first));
        }

        // 取出sort_map中前k个元素的value值(即单词)，放入到vector中，并返回结果
        vector<string> ret;
        for(auto& e : sort_map)
        {
            ret.push_back(e.second);
            if(--k == 0)
            {
                break;
            }
        }
        return ret;
    }
};

---