백준 / 18870번 좌표압축 / Python 파이썬

 

*문제 출처는 백준에 있습니다.

문제 제목: 좌표압축

문제 사이트: https://www.acmicpc.net/problem/18870

18870번: 좌표 압축

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

2 초

512 MB

65551

27437

20931

39.615%

문제

수직선 위에 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

출처

• 문제를 만든 사람: baekjoon

알고리즘 분류

• 정렬
• 값 / 좌표 압축

---

나의 풀이(첫풀이, index 이용, 시간초과)

import sys

input = sys.stdin.readline

n = int(input())
arr = list(map(int,input().split()))
sort_arr = sorted(set(arr))
for i in arr:
    print(sort_arr.index(i),end= " ")

 

두번째풀이 (해시이용, 통과)

import sys

input = sys.stdin.readline

n = int(input())
arr = list(map(int,input().split()))
sort_arr = sorted(set(arr))

dic = {sort_arr[i]:i for i in range(len(sort_arr))}
print(dic)
for i in range(len(arr)):
    print(dic[arr[i]],end=" ")

 

다른 풀이

나랑 비슷한데, 딕셔너리 대신 enumerate 를 이용해서 인덱싱하여 출력하였다

import sys
stdin = sys.stdin.buffer

stdin.readline()
arr = list(map(int, stdin.read().split()))
dic = {x: i for i, x in enumerate(sorted(set(arr)))}
print(' '.join(map(str, [dic[x] for x in arr])))

---