[BOJ] 18870번 "좌표 압축" 파이썬 풀이

문제

수직선 위에 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을 공백 한 칸으로 구분해서 출력한다.

---

입력 예시

5
2 4 -10 4 -9

6
1000 999 1000 999 1000 999

출력 예시

2 3 0 3 1

1 0 1 0 1 0

답안

코드

 

import sys

N = int(sys.stdin.readline().strip())

a = list(map(int, sys.stdin.readline().strip().split()))

mid = sorted(a)[len(a)//2]

b = sorted(set(a))

def binary_search(arr, target):
    left = 0
    right = len(arr) - 1

    while left <= right:
        mid = (left + right) // 2

        if arr[mid] == target:
            return mid
        elif arr[mid] < target:
            left = mid + 1
        else:
            right = mid - 1

    return -1

for num in a:
    index = binary_search(b, num)
    print(index, end=' ')

풀이