백준 [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

n = int(sys.stdin.readline())
numbers = list(sys.stdin.readline().split())
numbers = list(map(int, numbers))
numbers2 = list(set(numbers))

new = []
for i in range(n):
    cnt = 0    
    for j in range(len(numbers2)):
        if numbers[i] > numbers2[j]:
            cnt += 1
    new.append(cnt)

for k in new:
    print(k, end=' ')

정답은 맞추었으나 시간 초과로 제출이 거부되었다. 이중 for문을 사용한 것이 문제가 된 것으로 보인다.

• 수정한 코드

import sys

n = int(sys.stdin.readline())
numbers = list(map(int, input().split()))
numbers2 = list(set(numbers))

numbers2.sort()

new ={numbers2[i]:i for i in range(len(numbers2))}

for i in numbers:
    print(new[i], end=' ')

dictionary 자료형을 이용하니 시간 복잡도가 줄어들었다.