1. 程式人生 > >LeetCode——中級演算法——排序和搜尋——前K個高頻元素(JavaScript)

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; };```