백준 2110 공유기 설치(매개변수탐색)

문제

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

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

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

입력

첫째 줄에 집의 개수 N (2 ≤ N ≤ 200,000)과 공유기의 개수 C (2 ≤ C ≤ N)이 하나 이상의 빈 칸을 사이에 두고 주어진다. 둘째 줄부터 N개의 줄에는 집의 좌표를 나타내는 xi (0 ≤ xi ≤ 1,000,000,000)가 한 줄에 하나씩 주어진다.

출력

첫째 줄에 가장 인접한 두 공유기 사이의 최대 거리를 출력한다.

예제 입력 1 복사

5 3
1
2
8
4
9

예제 출력 1 복사

3

 

 

풀이

 

가장 인접, 가장 먼 등의 키워드가 들어오면 매개변수탐색일 가능성이 높다

N개의 집 중 C개의 집을 골라서 설치할때 M의 거리로 배치 가능한 갯수 D가 얼마인지를 본다
C <= D면 거리를 더 늘려서 볼 수 있고 C > D면 거리를 좁혀야 한다

 

처음에 GetCount를 무조건 바로 인접한 곳에서 그 전에것을 빼는 식으로 생각하다가 생각해보면

1,2번은 거리가 작아서 불가능하지만 1,3번은 가능할 수도 있다는걸 알게 되었다.

맨 처음 위치를 변수에 저장해두고 배치 가능한 거리가 나오면  변수에 현재 인덱스 위치를 저장해주는 식으로

GetCount 함수를 변경했다.. 기초적인 실수 같은데 이런건 하지 않게 유의하자

 

 static boolean GetCount(long[] arr, int count, long width){
        //처음에 위치 비교를 위해 0번에 박앗으니 1개 가지고 시작
        int tmp = 1;
        long past = arr[0];
        for (int i = 1; i < arr.length; i++){
            if (arr[i] - past >= width) {
                past = arr[i];
                tmp++;
            }
        }
        return tmp >= count;
    }

 

풀이 코드

class Main
{
    static long Solution(long[] arr, int count) {
        long l = 0, r = arr[arr.length-1], ans = -1;
        while (l <= r) {
            long m = (l + r) / 2;
            if (GetCount(arr,count,m)){
                ans = Math.max(m, ans);
                l = m+1;
            }
            else r = m-1;
        }
        return ans;
    }

    static boolean GetCount(long[] arr, int count, long width){
        //처음에 위치 비교를 위해 0번에 박앗으니 1개 가지고 시작
        int tmp = 1;
        long past = arr[0];
        for (int i = 1; i < arr.length; i++){
            if (arr[i] - past >= width) {
                past = arr[i];
                tmp++;
            }
        }
        return tmp >= count;
    }


    public static void main(String[] args) throws IOException {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int m = in.nextInt();

        long[] arr = new long[n];
        for (int i = 0; i < n; i++) arr[i] = in.nextLong();
        Arrays.sort(arr);

        System.out.println(Solution(arr, m));
    }
}