LeetCode——中級演算法——排序和搜尋——前K個高頻元素(JavaScript)
阿新 • • 發佈:2018-12-06
給定一個非空的整數陣列,返回其中出現頻率前 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;
};```