[Python] 백준 #18870. 좌표 압축

https://www.acmicpc.net/problem/18870

18870번: 좌표 압축

 

# 18870 - Python 3

import sys

input = sys.stdin.readline

n = int(input())

arr = list(map(int, input().split()))

new_arr = list(set(arr))                            # 중복 제거한 arr를 새로운 list인 new_arr에 저장
new_arr.sort(reverse = False)

dic = {new_arr[i] : i for i in range(len(new_arr))} # 시간복잡도를 줄이기 위하여 dictionary 사용
    
for i in arr:
    print(dic[i], end=" ")

< line 12 >

- dic = {new_arr[i] : i for i in range(len(new_arr))}

: key를 new_arr[i], value를 i(크기 순서로 나열했을 때, 그 순서)로 하는 dictionary 생성

: 반복문 사용시보다 시간 줄일 수 있음