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

가장 작은수부터 0 > 1  > 2 이런식으로 치환하면 되는 문제이다.

 

그래서 정렬및 중복제거를 한 리스트를 만든다음에, 그 리스트값의 인덱스값을 넘겨주는 방식으로 만들면 될 꺼라 생각했었다.

 

import sys
N = int(sys.stdin.readline())
lst = list(map(int,sys.stdin.readline().split()))
s_lst = sorted(set(lst))
result_lst = []
for i in lst:
    result_lst.append(s_lst.index(i))
for j in result_lst:
    print(j,end=' ')

그런데 시간초과가 떴다.. 왜일까 고민을 하다보니 index함수는 찾을때마다 모든리스트를 다 도는 방식의 함수라고 한다. 그래서 시간복잡도를 줄이기 위해서 딕셔너리를 사용해보았다. 코드 구성은 똑같다.

 

import sys
N = int(sys.stdin.readline())
lst = list(map(int,sys.stdin.readline().split()))
s_lst = sorted(set(lst))
dic = {s_lst[i] : i for i in range(len(s_lst))}
for j in lst:
    print(dic[j],end=' ')

딕셔너리를 이용하니 시간복잡도 문제가 해결되었다! 무언가 대응하는 관계를 만들때는 딕셔너리를 이용하는 습관을 들이는게 좋을 것 같다.