https://www.acmicpc.net/problem/2110
문제
- N개의 집이 있고, 좌표가 주어진다.
- C개의 공유기를 설치하여 최대한 많은 곳에서 와이파이를 사용하려고 한다.
- 공유기는 한 집에 하나만 설치할 수 있다.
- 가장 인접한 두 공유기 사이의 거리를 가능한 크게 하여 설치한다.
출력
가장 인접한 두 공유기 사이의 최대 거리를 출력한다.
풀이
- 거리를 이분 탐색하면 된다.
- 시작할 때 최소 거리는 바로 옆집이므로 1 이고, 최대 거리는 주어진 좌표의 끝과 끝이다.
- 현재 집에서 중간 거리 ((최소 + 최대) // 2) 보다 큰 집에 공유기를 설치한다.
- 중간 거리에 대해 공유기를 설치한 것이 보유한 공유기 (C) 보다 더 많은 경우, 거리를 더 줄여야 한다.
반대로 공유기를 더 적게 사용한 경우 거리를 더 늘려도 된다.
코드
from sys import stdin
input = stdin.readline
N, C = map(int, input().split())
house = [int(input()) for _ in range(N)]
house.sort()
start, end = 1, (house[-1] - house[0])
result = 0
while start <= end:
mid = (start + end) // 2
cur = house[0]
count = 1
for i in range(1, N):
if house[i] >= cur + mid:
count += 1
cur = house[i]
if count >= C:
start = mid + 1
result = mid
else:
end = mid - 1
print(result)
'알고리즘 공부 > 이진 탐색' 카테고리의 다른 글
[백준/이진탐색] 2512번 예산 (0) | 2024.06.21 |
---|