【LeetCode】692. 前K个高频单词

前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 次

思路

• 这个题目看到统计次数，首先想到的是使用哈希表。用哈希表来统计每个单词出现的频率。然后根据频率降序排列。输出前k个。
• 题目中的排序法则：按单词出现频率由高到低排序；不同的单词出现相同的频率，按字母顺序排序。规定排序法则，首先想到的是sort.Slice(）这个方法。
• 字符串的比较方法可以直接是a<b。

解法

func topKFrequent(words []string, k int) []string {
    hashMap:=map[string]int{}
    for _,v:=range words{ // 哈希表存放每个单词出现的频率
        hashMap[v]++
    }
    res:=[]string{}
    for k,_:=range hashMap{ // 存放哈希表中存在的单词
        res=append(res,k)
    }
    
    sort.Slice(res,func(i,j int)bool{ // 定义比较规则
        return hashMap[res[i]]>hashMap[res[j]]||hashMap[res[i]]==hashMap[res[j]]&&res[i]<res[j]
    })
    return res[:k]
}