코드 수정

- longestConsecutive - bucket sort 사용
- top-k-frequent-elements - 코드 최적화

Author: seungseung88

longest-consecutive-sequence/seungseung88.js

@@ -18,12 +18,11 @@ const longestConsecutive = (nums) => {
     // 일단 연속되는 배열에서는 가장 최솟값을 찾음
     if (!set.has(v - 1)) {
       let cnt = 1;
-
       // 연속되는 배열에서 가장 긴 배열을 저장
       while (set.has(v + cnt)) {
         cnt += 1;
-        longest = longest < cnt ? cnt : longest;
       }
+      longest = Math.max(longest, cnt);
     }
   });
 

top-k-frequent-elements/seungseung88.js

@@ -1,32 +1,46 @@
 /**
- * 시간 복잡도: O(n log n)
- * - nums 배열 순회하며 빈도수 카운트: O(n)
- * - 빈도수 기준 내림차순 정렬: O(n log n)
- * - 상위 k개 선택: O(k) -> k는 n보다 작으므로 무시
+ * 시간 복잡도: O(n)
+ * - nums 순회하며 map 생성 O(n)
+ * - map 순회하며 bucket 생성 O(n)
+ * - bucket 순회하며 결과 찾음 O(n)
  *
  * 공간 복잡도: O(n)
- * - countNums 객체: 최악의 경우 모든 숫자가 다른 경우 O(n)
- * - sortedCountNums 배열: countNums와 동일한 크기 O(n)
- * - answer 배열: k 크기이지만 k는 n보다 작으므로 무시
+ * - map O(n)
+ * - bucket O(n)
+ * - result k
  */
 const topKFrequent = (nums, k) => {
-  // 각 숫자의 빈도수를 저장하는 객체
-  const countNums = {};
+  // nums 요소 : 요소의 갯수
+  const map = {};
+  // 요소의 갯수 : Set{ nums 요소 }
+  const bucket = [];
+
+  const result = [];
 
-  // nums 배열을 순회하며 각 숫자의 빈도수를 카운트
   for (let i = 0; i < nums.length; i += 1) {
-    const num = nums[i];
-    countNums[num] = !countNums[num] ? 1 : countNums[num] + 1;
+    if (map[nums[i]]) {
+      map[nums[i]] += 1;
+    } else {
+      map[nums[i]] = 1;
+    }
   }
 
-  // 빈도수를 기준으로 내림차순 정렬
-  const sortedCountNums = Object.entries(countNums).sort((a, b) => b[1] - a[1]);
-  const answer = [];
+  for (let [num, freq] of Object.entries(map)) {
+    if (bucket[freq]) {
+      bucket[freq] = bucket[freq].add(num);
+    } else {
+      bucket[freq] = new Set().add(num);
+    }
+  }
 
-  // 상위 k개의 숫자를 answer 배열에 저장
-  for (let i = 0; i < k; i += 1) {
-    answer[i] = Number(sortedCountNums[i][0]);
+  for (let i = bucket.length - 1; i >= 0; i -= 1) {
+    if (bucket[i]) {
+      // string을 number로 변환
+      const value = Array.from(bucket[i], Number);
+      result.push(...value);
+    }
+    if (result.length === k) break;
   }
 
-  return answer;
+  return result;
 };