[알고리즘]/문제 풀이

백준 18870번 : 좌표 압축

JuJu(INTJ) 2023. 10. 18. 14:13

문제

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

Xi를 좌표 압축한 결과 X'i의 값은 Xi > Xj를 만족하는 서로 다른 좌표 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

 

 

풀이

1. 가장 작은 수의 인덱스는 0으로 시작

2. 그 이후에 인덱스를 구해서 작은 순서대로 값을 매겨준다.

N = int(input())
x_list = list(map(int, input().split()))

# 좌표 압축을 수행할 때 사용할 딕셔너리 생성
coord_compression = {}

# 입력된 좌표를 정렬하여 중복을 제거한 리스트를 생성
sorted_x_list = sorted(set(x_list))

# 딕셔너리를 이용하여 좌표 압축 수행
for i, x in enumerate(sorted_x_list):
    coord_compression[x] = i

# 좌표 압축 결과를 출력
for x in x_list:
    print(coord_compression[x], end=' ')
반응형