[백준/Python] 18870번 좌표 압축

1️⃣ 문제 설명

수직선 위에 N개의 좌표 X1, X2, ..., XN이 있다. 이 좌표에 좌표 압축을 적용하려고 한다.

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

X1, X2, ..., XN에 좌표 압축을 적용한 결과 X'1, X'2, ..., X'N를 출력해보자.

2️⃣ 입력

첫째 줄에 N이 주어진다.

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

3️⃣ 출력

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

 

---

👩🏻‍💻 작성한 코드

import sys
import heapq

input = sys.stdin.readline

n = int(input())
pos = list(map(int, input().split()))
setP = sorted(list(set(pos))) # 중복값 없애주기 위해 set으로 변환 후, 정렬된 리스트로 변환
dic = {}	# setP 인덱스 정보 저장할 딕셔너리

for i in range(len(setP)):
    dic[setP[i]]=i

for i in pos:
    print(dic[i], end = ' ')

💡 코드 설명

이 문제는 리스트 안에서 원소의 크기 순서를 출력하면 된다.

 

중복되는 원소의 크기 순서는 서로 같으므로 우선 set으로 변환하여 중복값을 제거해준 뒤, 리스트로 다시 변환시켜 정렬해주었다.

for i in range(len(setP)):
    dic[setP[i]]=i

⬆️ for문을 통해, 딕셔너리 dic에 { setP[ i ] : i } 로 key값과 value값을 저장해준다.

 

예제 1을 입력했을 때, 아래 사진과 같이 dic에 저장되는 것을 확인할 수 있다.

for i in pos:
    print(dic[i], end = ' ')

⬆️ for문을 통해, dic에서 i와 매핑되는 value값을 출력해준다.