[백준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- 시간초과가 나왔다. index함수가 시간을 많이 사용한다. 다른 방법을 사용해야한다.

import sys
input=sys.stdin.readline

n=int(input())
x=list(map(int,input().split()))
tmp=sorted(x)
ans=[0 for i in range(n)]
j=-1
prev='-'
for i in range(n):
    index=x.index(tmp[i])
    x[index]='-'
    if prev!=tmp[i]:
        j=j+1
    ans[index]=j
    prev=tmp[i]
for i in ans:
    print(i,end=' ')

2-set을 이용하여 중복을 제거하고 딕셔너리를 이용해 순서를 넣어주었다.

import sys
input=sys.stdin.readline

n=int(input())
x=list(map(int,input().split()))
tmp=sorted(set(x))
dic={}
for i in range(len(tmp)):
    dic[tmp[i]]=i
for i in x:
    print(dic[i],end=' ')