[Python] 백준 18870번 좌표 압축

문제

수직선 위에 N개의 좌표 X1, X2, ..., XN이 있다. 이 좌표에 좌표 압축을 적용하려고 한다.
Xi를 좌표 압축한 결과 X'i의 값은 Xi > Xj를 만족하는 서로 다른 좌표 Xj의 개수와 같아야 한다.
X1, X2, ..., XN에 좌표 압축을 적용한 결과 X'1, X'2, ..., X'N를 출력해보자.

 

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

18870번: 좌표 압축

입력

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

출력

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

제한

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

예제

# input
5
2 4 -10 4 -9
# output
2 3 0 3 1

# input
6
1000 999 1000 999 1000 999
# output
1 0 1 0 1 0

문제 풀이

# 1 (오답)
import sys

N = int(sys.stdin.readline())
L = list(map(int,sys.stdin.readline().split()))
ans = [0] * N

for i in range(N):
    for j in range(N):
        if L[i] > L[j]:
            ans[i] += 1
print(ans)      

# 2
import sys

N = int(sys.stdin.readline())
L = list(map(int,sys.stdin.readline().split()))

ans = sorted(list(set(L)))
D = {ans[i]:i for i in range(len(ans))}

for i in L:
    print(D[i])

처음에는 전체 입력값을 대조하는 식으로 풀었으나, 시간 초과로 인해 오답 처리 되었습니다.

 

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.