알고리즘 도전기 - 75

김치전사·2022년 7월 18일
0

알고리즘 도전기

목록 보기
73/89

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

예제 입력

5
2 4 -10 4 -9

예제 출력

2 3 0 3 1

코드

import sys


def binary_search(target, low, high):
    if low > high:
        return -1

    mid = (low + high) // 2
    if sort_arr[mid] == target:
        return mid
    elif sort_arr[mid] > target:
        return binary_search(target, low, mid - 1)
    else:
        return binary_search(target, mid + 1, high)


N = int(sys.stdin.readline().rstrip())
arr = list(map(int, sys.stdin.readline().split()))
sort_arr = list(set(arr))

sort_arr.sort()
result_arr = []
for e in arr:
    idx = binary_search(e, 0, len(sort_arr) - 1)
    result_arr.append(idx)

print(*result_arr)
profile
개인공부 블로그입니다. 상업적 용도 X

0개의 댓글