【路飞】 692. 前K个高频单词 js实现

83 阅读1分钟

上题目:

给一非空的单词列表,返回前 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 总为有效值, 1 ≤ k ≤ 集合元素数。 输入的单词均由小写字母组成。  

解题:

/**
 * @param {string[]} words
 * @param {number} k
 * @return {string[]}
 */
var topKFrequent = function(words, k) {
    var hash = {};
    for (var i = 0; i < words.length; i++) {
        var c = words[i];
        hash[c] = hash[c] ? hash[c] + 1 : 1;
    }
    var array = Object.entries(hash).sort(function(a, b) {
        if (a[1] > b[1]) {
            //a,b都是数组
            return -1;
        } else if (a[1] < b[1]) {
            return 1;
        } else {
            var tem = a[0].localeCompare(b[0]);
            if (tem > 0) {
                return 1;
            } else {
                return -1;
            }
        }
    });
    var res = [];
    for (var j = 0; j < k; j++) {
        res.push(array[j][0]);
    }
    return res;
};