[Inflearn] 자바 알고리즘 문제풀이 #06-10 9. 마구간 정하기(결정 알고리즘)

문제

설명

N개의 마구간이 수직선상에 있습니다. 각 마구간은 x1, x2, x3, ......, xN의 좌표를 가지며, 마구간간에 좌표가 중복되는 일은 없습니다.

현수는 C마리의 말을 가지고 있는데, 이 말들은 서로 가까이 있는 것을 좋아하지 않습니다. 각 마구간에는 한 마리의 말만 넣을 수 있고,

가장 가까운 두 말의 거리가 최대가 되게 말을 마구간에 배치하고 싶습니다.

C마리의 말을 N개의 마구간에 배치했을 때 가장 가까운 두 말의 거리가 최대가 되는 그 최대값을 출력하는 프로그램을 작성하세요.

 

입력 

첫 줄에 자연수 N(3<=N<=200,000)과 C(2<=C<=N)이 공백을 사이에 두고 주어집니다.

둘째 줄에 마구간의 좌표 xi(0<=xi<=1,000,000,000)가 차례로 주어집니다.

 

출력 

첫 줄에 가장 가까운 두 말의 최대 거리를 출력하세요.

예시 입력 1 

5 3
1 2 8 4 9

예시 출력 1

3

 

 

 

선생님 풀이

import java.util.Arrays;
import java.util.Scanner;

public class Main6_10 {
    public int count(int[] arr, int dist) {
        int cnt=1; // cnt: 배치한 말의 마리 수, 1마리 배치했다
        int ep=arr[0]; // ep: 바로 전에 배치한 말의 위치, 첫 번째 말을 맨 왼쪽 위에 놓았다고 가정
        for (int i = 1; i < arr.length; i++) {
            if(arr[i] - ep >= dist) {
                cnt++;
                ep=arr[i];
            }
        }
        return cnt;
    }

    public int Solution(int n, int c, int[] arr) {
        int answer=0;
        int lt = 1, rt=arr[n-1];
        Arrays.sort(arr);

        while(lt<=rt) {
            int mid = (lt+rt)/2;
            if(count(arr, mid)>=c) { // c마리 수 이상 배치 ==> 유효
                answer=mid;
                lt=mid+1; // 간격을 넓혀서 최대 값을 찾아봐야지
            }
            else rt = mid-1; // 간격을 좁혀서 c마리 모두 배치될 수 있게 해야지
        }
        return answer;
    }

    public static void main(String[] args) {
        Main6_10 T = new Main6_10();
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int c = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        System.out.println(T.Solution(n, c, arr));
    }
}

결정 알고리즘 적응하기 왜이렇게 어렵늬,,,,, count함수(유효 판단 함수) 생각하는게 진짜 안되는 듯

내일은 복습해야지 !_!

 

 

 

결과