给一非空的单词列表,返回前 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 次。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-words
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
vector<string> topKFrequent(vector<string>& words, int k) {
map<string,int> mp;
vector<string> res;
for (auto& item : words)
{
mp[item]++;
}
//由于map不支持value的sort排序,所以必须将map转换为vector才能进行sort自定义排序
vector<pair<string, int>> vec(mp.begin(), mp.end());
//sort的第三个参数是lambda表达式,其实和传普通函数一样,sort的第三个参数是支持二元谓词的
sort(vec.begin(), vec.end(), []
(const pair<string,int> &left, const pair<string,int> &right)
{
if (left.second == right.second)
return left.first < right.first;
else
return left.second > right.second;
});
//将vector的前k个元素添加到res中
for (const auto &item : vec)
{
if (k-- == 0)
break;
res.push_back(item.first);
}
return res;
}