[路飞]_程序员必刷力扣题: 692. 前K个高频单词

113 阅读1分钟

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, 21 次。

提示:

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

计数+排序

思路 根据题意我们首先要找到出现次数最多的单词,如果出现次数相同则按照英文字母排序

  • 遍历统计每个单词出现的次数,用一个对象
  • 对对象的keys数组进行排序,规则按照如果出现次数不同,则直接降序排序,如果次数相同,则直接用localeCompare这个api来比较两个单词的前后关系 a.localeCompare(b)中如果a中的单词ascll码比较靠前则返回-1,相同返回0,靠后返回1

例:

'a'.localeCompare('b')
//-1

排序完毕后我们直接截取keys中0到k位置的单词即可

var topKFrequent = function (words, k) {
    var map = {}
    for (var i = 0; i < words.length; i++) {
        var item = words[i]
        if (map[item]) {
            map[item]++
        } else {
            map[item] = 1
        }
    }
    var keys = Object.keys(map)

    keys.sort((a, b) => {
        if (map[b] !== map[a]) {
            return map[b] - map[a]
        }
        return a.localeCompare(b)
    })
    return keys.slice(0, k)
};