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

풀이

​

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

​

라는 조건을 보고 정렬 후 원소의 인덱스 값이 압축한 좌표와 같다는 것을알게 되었다.

무슨 말이냐, [ 1, 3, 2, 4, 5] 이런 배열에서 1을 압축하면 1보다 작은 좌표의 개수는 없기 때문에 0

3을 압축하면 3보다 작은 좌표의 개수는 [1, 2] 로 2개가 되므로 3, 즉 정렬 했을 때 [ 1, 2, 3, 4, 5]의 인덱스 값이 해당 좌표의 압축한 값이다.

​

import sys

n = int(sys.stdin.readline().replace("\n", ""))
l = list(map(int, sys.stdin.readline().replace("\n", "").split()))

r = list(sorted(set(l)))

d = {value: index for index , value in enumerate(r)}


for i in l:
    print(d[i],end=' ')

enumerate()를 이용해 Pythonic한 코드를 짰다.