/
topk.js
41 lines (33 loc) · 846 Bytes
/
topk.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
/**
* @param {number[]} nums
* @param {number} k
* @return {number[]}
*/
// Solution write up https://leetcode.com/problems/top-k-frequent-elements/solutions/3583449/top-k-frequent-elements/
const topKFrequent = (nums, k) => {
const numMap = new Map();
nums.forEach((element) => {
if (numMap.has(element)) {
let count = numMap.get(element);
count++;
numMap.set(element, count);
} else {
numMap.set(element, 1);
}
});
const sortedNumMap = [...numMap].sort((a, b) => {
if (a[1] < b[1]) {
return 1;
} else {
return -1;
}
});
const answer = [];
for (let i = 0; i < k; i++) {
answer.push(sortedNumMap[i][0]);
}
return answer;
};
let nums; let k;
nums = [1,1,1,2,2,3]; k = 2;
console.log(nums, k, topKFrequent(nums, k));