学习记录：js算法（四）： 前 K 个高频元素

前 K 个高频元素

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:
输入: nums = [1], k = 1
输出: [1]

思路一

// 正常写法
var topKFrequent = function (nums, k) {
   var topKFrequent = function (nums, k) {
    let arr = []
    let obj = {}
    for (let num of nums) {
        obj[num] = obj[num] ? obj[num] + 1 : 1
    }
    for (let key in obj) {
        if (obj[key] >= k) {
            arr.push(key)
        }
    }
    return arr
};
// 花里花哨写法
var topKFrequent = function (nums, k) {
    let obj = {};
    for (let num of nums) {
        obj[num] = (obj[num] || 0) + 1; // 如果存在则加1，不存在则初始化为1
    }
    return Object.entries(obj)
        .filter(([key, value]) => value >= k)
        .map(([key]) => key);
};

思路二

// 正常写法
var topKFrequent = function (nums, k) {
    let arr = []
    let map = new Map()
    for (let i in nums) {
        let key = nums[i]
        let count = map.get(key) ? map.get(key) : 0
        map.set(key, count + 1)
    }
    for (let [key, value] of map) {
        if (value >= k)
            arr.push(key)
    }
    return arr
};
// 花里胡哨的写法
var topKFrequent = function (nums, k) {
    const map = new Map();
    for (const num of nums) {
        map.set(num, (map.get(num) || 0) + 1);
    }
    return [...map.entries()]
        .filter(([key, value]) => value >= k)
        .map(([key]) => key);
};