백준: 2110 공유기 설치

문제

2110번: 공유기 설치

 

문제 풀이

 앞서 다룬, 랜선 자르기, 나무 자르기와 동일한 이분 탐색 유형의 문제이다. N개의 집이 주어 질 때, 각각의 집 사이에 C개의 공유기를 적절히 설치하는 최대 거리를 반환하는 문제이다. 예제 입력 1의 경우 1 2 4 8 9와 같이 집이 위치한다. 공유기 3대를 최대 거리로 설치하기 위해서는 3이라는 것을 알 수 있다. 이를 이분 탐색으로 찾기 위해서는 1과 마지막 집을 기준으로 이분 탐색을 과정을 반복하면 된다.

 

코드

from sys import stdin


if __name__ == "__main__":
    n, c = map(int, stdin.readline().split())
    locations = sorted([int(stdin.readline()) for _ in range(n)])
    s, e = 1, locations[-1] - locations[0]

    while s <= e:
        set_loc = 0
        m = (s + e) // 2
        set_cnt = 1

        for i in range(1, n):
            if locations[i] >= locations[set_loc] + m:
                set_cnt += 1
                set_loc = i

        if set_cnt >= c:
            s = m + 1
        else:
            e = m - 1

    print(e)

 이전의 문제들과 다른 점이 있다면 하나의 집을 선택하고, 거리 (mid) 만큼 차이가 나는지 확인하고 이를 반복하면 답을 찾을 수 있다.