leetcode 692. 前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 次。

注意：

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

【思路分析】

和703题求第K大元素的思路类似，这是典型的可以用优先队列来解的题。不过和数字不同，字母的比较需要我们自己写一个函数。结合我之前写的C++中的优先队列的:https://blog.csdn.net/zpznba/article/details/84593107.可以写出如下解法：

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        typedef pair<string,int>node;
        unordered_map<string,int> m;
        vector<string> res(k);
        struct cmp
        {
            bool operator()(node& a, node& b) {
                if(a.second == b.second)
                    return a.first<b.first;
                return a.second>b.second;
            }
        };
        priority_queue<node,vector<node>,cmp> q;
        
        for(auto c :words){
            ++m[c];
        }
        for(auto it : m){
            q.push(it);
            if(q.size()>k)
                q.pop();
        }
        for(int i=k-1;i>=0;--i){
            res[i]=q.top().first;
            q.pop(); 
        }
        return res;
    }
};

如果单词出现次数不同，则出现次数少的排在前面（小顶堆思想）；若相同，则按字符串的顺序，把最大的放在堆顶（大顶堆思想）。另外，在往res中存的时候也要注意顺序问题。感觉用了高级数据结构反而写的很慢，各种出错。可能是因为还不熟悉的原因吧。于是我直接用排序的方法来做一遍：

typedef pair<string,int> node;
bool compare(node& a, node& b) {
    if(a.second == b.second) {
        return a.first < b.first;
    }
    return a.second > b.second;
}

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        vector<string> result;
        unordered_map<string, int> m;
        for(string word: words) {
            m[word]++;
        }
        vector<node> tmp;
        for(auto val: m) {
            tmp.push_back(val);
        }
        sort(tmp.begin(), tmp.end(), compare);
        for(int i = 0; i < k; i++) {
            result.push_back(tmp[i].first);
        }
        return result;
    }
};