LeetCode——中级算法——排序和搜索——前K个高频元素(JavaScript)

给定一个非空的整数数组,返回其中出现频率前 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;
};```

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-02 17:28
25届每天都在焦虑找工作的事情0offer情绪一直很低落硬撑着面了一个岗位岗位有应酬的成分面试的时候hr给我出各种场景题问的问题比较犀利&nbsp;有点压力面的感觉感觉有点回答不上来本来就压抑的情绪瞬间爆发了呢一瞬间特别想哭觉得自己特别没用没绷住掉眼泪了事后想想觉得自己挺有病的&nbsp;真的破大防了
喜欢唱跳rap小刺猬...:我觉得没关系吧,之前有一次面试leader给我压力面,我顶住了压力,结果入职的时候发现组里氛围很差,果断跑路。其实从面试就能大概看出组的情况,面试体验好的组倒是不一定好,但是面试体验不好的组。。。就很难说
面试尴尬现场
点赞 评论 收藏
分享
06-17 21:57
门头沟学院 Java
白友:噗嗤,我发现有些人事就爱发这些,明明已读不回就行了,就是要恶心人
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务