Lt347. 前 K 个高频元素

347. 前 K 个高频元素

给定一个非空的整数数组，返回其中出现频率前 _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 *是数组的大小。
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
• 你可以按任意顺序返回答案。

思路

哈希表排序

借助哈希表记录元素个数，sort 排序，取前 k 个元素返回。时间复杂度最大的为排序，O(nlogn)。

解答

哈希表排序

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var topKFrequent = function (nums, k) {
  const map = new Map();
  for (const num of nums) map.set(num, map.has(num) ? map.get(num) + 1 : 1);
  return [...map.entries()]
    .sort((a, b) => b[1] - a[1])
    .splice(0, k)
    .map((e) => e[0]);
};