백준 2110 자바 - 공유기 설치

https://www.acmicpc.net/problem/2110

2110번: 공유기 설치

문제

도현이의 집 N개가 수직선 위에 있다. 각각의 집의 좌표는 x1, ..., xN이고, 집 여러개가 같은 좌표를 가지는 일은 없다.

도현이는 언제 어디서나 와이파이를 즐기기 위해서 집에 공유기 C개를 설치하려고 한다. 최대한 많은 곳에서 와이파이를 사용하려고 하기 때문에, 한 집에는 공유기를 하나만 설치할 수 있고, 가장 인접한 두 공유기 사이의 거리를 가능한 크게 하여 설치하려고 한다.

C개의 공유기를 N개의 집에 적당히 설치해서, 가장 인접한 두 공유기 사이의 거리를 최대로 하는 프로그램을 작성하시오.

코드

import java.io.*;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    private static int[] list;
    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int c = Integer.parseInt(st.nextToken());

        list = new int[n];

        for (int i = 0; i < n; i++) {
            list[i] = Integer.parseInt(br.readLine());
        }
        Arrays.sort(list);

        int low = 1, high = list[n - 1] - list[0] + 1;

        while (low < high) {
            int middle = (high + low) / 2;
            if (numOfInstallable(middle) < c) {
                high = middle;
            } else {
                low = middle + 1;
            }
        }
        bw.write(String.valueOf(low - 1));
        bw.flush();
    }
    private static int numOfInstallable(int distance) {
        int count = 1;
        int last = list[0];

        for (int i = 1; i < list.length; i++) {
            int locate = list[i];
            if (locate - last >= distance) {
                count++;
                last = locate;
            }
        }
        return count;
    }
}

풀이

최소 거리를 지정하고, 최소 거리보다 짧으면 설치하지 않는 방식으로 주어진 개수의 공유기를 설치할 수 있는 최소 거리를 가장 크게 설정하면 된다. 입력 받아 정렬을 수행한 후, 이분 탐색을 돌리는데, numOfInstallable 함수에서 몇 개의 공유기를 설치 가능한지 구하면서 while문을 돌린다. 이후 low 에 1을 빼 출력한다. Upper Bound 형식으로의 탐색은 만족하는 값보다 1 큰 경우를 찾기 때문이다.