2022.01.25 33일차

백준 2110번 (공유기 설치)

문제

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

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

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

나의 풀이

1. 가장 인접한 공유기 사이의 최대 길이, 이진탐색
2. 집의 개수 N, 공유기의 개수 C, 집의 좌표
3. 구해야 되는 값 min, max, mid, count, answer
4. 처음 이진탐색의 min, max 조건은 1과 배열의 최대값
5. 집 사이의 거리(최대 길이)를 만족할 때마다 카운트
6. 카운트(count) >= 공유기 개수(C) 충족하면 answer = mid, min값 바꾸기 min = mid + 1
7. 카운트(count) < 공유기 개수(C) 이면 max값 변경
8. 이때 집 사이의 거리는 최소 1, while문 조건이 min < max
9. max값은 max = mid로 변경

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        int C = scanner.nextInt();

        int min = 1;
        int max = Integer.MIN_VALUE;

        int[] nArr = new int[N];

        for (int i = 0; i < N; i++) {
            nArr[i] = scanner.nextInt();

            max = Math.max(nArr[i], max);
        }

        int answer = 1;
        Arrays.sort(nArr);

        while (min < max) {
            int count = 1;
            int mid = (min + max) / 2;

            int val = nArr[0];

            for (int x = 1; x < N; x++) {
                if (nArr[x] - val >= mid) {
                    count++;
                    val = nArr[x];
                }
            }

            if (count >= C) {
                answer = mid;
                min = mid + 1;
            } else {
                max = mid;
            }
        }

        System.out.println(answer);
    }
}

생각하기

• 이진탐색 시 시작, 끝 부분 조건 잘 파악하기