给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

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

示例 2:

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

说明：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

思路

第一步，记录每个数出现的次数

  let map = new Map();
  for (let num of nums) { // 记录每个数出现的次数
    map.set(num, (map.get(num) || 0) + 1);
  }
  

第二步，设置一个数组 countOfNum，其第 i 个元素表示出现 i 次的元素数组

  let countOfNum = Array(nums.length);
  for (let key of map.keys()) {
    let value = map.get(key);
    if (countOfNum[value] === undefined) {
      countOfNum[value] = [key]
    } else {
      countOfNum[value].push(key)
    }
  }

第三步，从 countOfNum 末尾开始往头遍历，将非空元素都加入结果数组中，直到结果数组的大小等于K。

  let res = [];
  for (let i = countOfNum.length - 1; i >= 0 && res.length < k; i--) {
    if (countOfNum[i] !== undefined) {
      res = res.concat(countOfNum[i])
    }
  }
  return res;

总代码

/** * @param {number[]} nums * @param {number} k * @return {number[]} */
var topKFrequent = function(nums, k) {
  let map = new Map();
  for (let num of nums) { // 记录每个数出现的次数
    map.set(num, (map.get(num) || 0) + 1);
  }
  
  let countOfNum = Array(nums.length);
  for (let key of map.keys()) {
    let value = map.get(key);
    if (countOfNum[value] === undefined) {
      countOfNum[value] = [key]
    } else {
      countOfNum[value].push(key)
    }
  }
  
  let res = [];
  for (let i = countOfNum.length - 1; i >= 0 && res.length < k; i--) {
    if (countOfNum[i] !== undefined) {
      res = res.concat(countOfNum[i])
    }
  }
  return res;
};```