347.前k个高频元素

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

示例 1:

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

示例 2:

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

提示:

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

进阶:你所设计算法的时间复杂度 必须 优于 O(n log n) ,其中 n 是数组大小。

方法一:哈希表+sort

时间复杂度:O(m+(nm)×Σ)

空间复杂度:O(Σ)

复制代码
 1 /**
 2  * @param {number[]} nums
 3  * @param {number} k
 4  * @return {number[]}
 5  */
 6 var topKFrequent = function(nums, k) {
 7     let map = new Map();
 8     for (let num of nums) {
 9         map.set(num, map.has(num) ? map.get(num) + 1 : 1) //初始化出现次数为1,之后累加
10     }
11     if (k === map.size) {
12         return [...map.keys()]; //k如果等于map.size,直接返回全部key
13     }
14     let arr = Array.from(map).sort((a, b) => { return b[1] - a[1] }); //从大到小排序
15     return arr.slice(0, k).map(n => n[0]); //截取前k个key
16 }
复制代码

 

posted @   icyyyy  阅读(16)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律
点击右上角即可分享
微信分享提示