[Baekjoon] 좌표압축

백준 #18870 좌표압축

문제

수직선 위에 N개의 좌표 X1, X2, ..., XN이 있다. 이 좌표에 좌표 압축을 적용하려고 한다.

Xi를 좌표 압축한 결과 X'i의 값은 Xi > Xj를 만족하는 서로 다른 좌표의 개수와 같아야 한다.

X1, X2, ..., XN에 좌표 압축을 적용한 결과 X'1, X'2, ..., X'N를 출력해보자.

입력

첫째 줄에 N이 주어진다.

둘째 줄에는 공백 한 칸으로 구분된 X1, X2, ..., XN이 주어진다.

출력

첫째 줄에 X'1, X'2, ..., X'N을 공백 한 칸으로 구분해서 출력한다.

제한

• 1 ≤ N ≤ 1,000,000
• -109 ≤ Xi ≤ 109

예제 입력 1 복사

5
2 4 -10 4 -9

예제 출력 1 복사

2 3 0 3 1

예제 입력 2 복사

6
1000 999 1000 999 1000 999

예제 출력 2 복사

1 0 1 0 1 0

 

---

✔️ 내가 쓴 코드

import sys
s=sys.stdin.readline
n=int(s())
arr=list(map(int,s().split()))
arr2=sorted(set(arr))

dic={v:i for i,v in enumerate(arr2)}
for i in arr:
  print(dic[i], end=' ')

▪️좌표가 같은 경우는 같은 인덱스값을 내보내야 하므로 set함수로 중복된 값을 제거해주어야 한다.

▪️value값이 중복된 값이 제거돼서 키값이 될 수 있다는 것을 이용해 enumerate로 index와 value를 동시에 받아 딕셔너리로 저장한다

❗️ 리스트에서 index찾는 식으로 하면 시간초과 뜸