算法题：用JS实现前 K 个高频元素

学习目标：

• 前 K 个高频元素
  leetcode原题链接

---

学习内容：

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

示例 1:

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

示例 2:

输入: nums = [1], k = 1
输出: [1]

提示：

• 1 <= nums.length <= 10^5
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

---

解题思路：
思路一 通过Set和Map
利用Set不包含重复项的特性，先设置一个去重后的数组arr，再创建一个map作为映射，遍历给定的数字数组nums，如果此时遍历的值不存在之前创建的map里，则将它加入到映射中，值为1，如果已经存在于映射中，则将之前映射的对应值加1。最后再通过sort方法对最开始创建的去重数组arr进行从大到小排序，排序方法就是映射中各个值的大小比较，最后，返回排序后的前k项即可得到最终结果。

实现代码：

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var topKFrequent = function (nums, k) {
    const arr = [...new Set(nums)]
    const map = new Map()

    nums.map(item => {
        if (map.has(item)) {
            map.set(item, map.get(item) + 1)
        } else {
            map.set(item, 1)
        }
    })

    return arr.sort((a, b) => map.get(b) - map.get(a)).slice(0, k)
} 

思路二 桶排序
实现方法就是先将数据划分到有限个桶里，再对桶进行排序，这里还是用map存储频次，用数组表达桶，频次作为数据下标，针对不同的频次的数字，聚合。

实现代码：

var topKFrequent = function (nums, k) {
    let map = new Map()
    let arr = [...new Set(nums)]

    nums.map(item => {
        if (map.has(item)) {
            map.set(item, map.get(item) + 1)
        } else {
            map.set(item, 1)
        }
    })

    if (map.size < k) {
        return [...map.keys()]
    }

    // 桶排序
    return bucket(map, k)
}

function bucket(map, k) {
    let res = []
    let arr = []
    
    map.forEach((value, key) => {
        // 可以利用频次作为下标，将数据分配到桶里
        if (!arr[value]) {
            arr[value]=[key]
        } else {
            arr[value].push(key)
        }
    })

    for (let i = arr.length - 1; i >= 0 && res.length < k; i--){
        if (arr[i]) {
            res.push(...arr[i])
        }
    }
    return res
}