[Silver II] Title: 좌표 압축, Time: 2920 ms, Memory: 248704 KB -BaekjoonHub

Author: codehyunn

백준/Silver/18870. 좌표 압축/README.md

@@ -4,15 +4,15 @@
 
 ### 성능 요약
 
-메모리: 281836 KB, 시간: 4788 ms
+메모리: 248704 KB, 시간: 2920 ms
 
 ### 분류
 
 값 / 좌표 압축, 정렬
 
 ### 제출 일자
 
-2024년 1월 26일 14:35:08
+2024년 1월 26일 14:42:40
 
 ### 문제 설명
 

백준/Silver/18870. 좌표 압축/좌표 압축.py

@@ -1,26 +1,17 @@
 import sys 
-import heapq
 from collections import defaultdict
 
 input = sys.stdin.readline
 
 n = int(input())
 nums = list(map(int,input().split()))
-answer = [0 for _ in range(n)]
 
-count = defaultdict(list)
-for idx, num in enumerate(nums) :
-    count[num].append(idx)
+idxs = defaultdict(list)
+for i in range(n) :
+    idxs[nums[i]].append(i)
 
-queue = []
-for key, values in count.items() :
-    heapq.heappush(queue, (key, values))
+for i, num in enumerate(sorted(set(nums))) :
+    for idx in idxs[num] :
+        nums[idx] = i
 
-total = 0
-while queue :
-    _, idxs = heapq.heappop(queue)
-    for idx in idxs :
-        answer[idx] += total  
-    total += 1
-
-print(*answer)
+print(* nums)