题目描述:
给一非空的单词列表,返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。
解题思路:
1. 采用map统计各个单词出现的次数;
2. 根据次数对map进行排序,注意相同出现次数相同时,按单词字典序排序;
3. 取出前 k 个出现次数最多的单词。
class Solution {
public:
vector<string> topKFrequent(vector<string>& words, int k) {
if(words.size()<k)
return {};
map<string, int> timesMap;
for(auto word : words){
timesMap[word]++;
}
vector<pair<string, int>> timesVec;
for(auto wordTimes : timesMap){
timesVec.push_back(wordTimes);
}
sort(timesVec.begin(), timesVec.end(), [](pair<string, int> a, pair<string, int> b){return a.second == b.second ? a.first < b.first : a.second > b.second;});
vector<string> ans;
for(int i=0; i<k; i++){
ans.push_back(timesVec[i].first);
}
return ans;
}
};