[이분탐색] 프로그래머스 징검다리 level4 자바 (java)

https://school.programmers.co.kr/learn/courses/30/lessons/43236

프로그래머스

문제 설명

출발지점부터 distance만큼 떨어진 곳에 도착지점이 있습니다. 그리고 그사이에는 바위들이 놓여있습니다. 바위 중 몇 개를 제거하려고 합니다.
예를 들어, 도착지점이 25만큼 떨어져 있고, 바위가 [2, 14, 11, 21, 17] 지점에 놓여있을 때 바위 2개를 제거하면 출발지점, 도착지점, 바위 간의 거리가 아래와 같습니다.

제거한 바위의 위치	각 바위 사이의 거리	거리의 최솟값
[21, 17]	[2, 9, 3, 11]	2
[2, 21]	[11, 3, 3, 8]	3
[2, 11]	[14, 3, 4, 4]	3
[11, 21]	[2, 12, 3, 8]	2
[2, 14]	[11, 6, 4, 4]	4

위에서 구한 거리의 최솟값 중에 가장 큰 값은 4입니다.

출발지점부터 도착지점까지의 거리 distance, 바위들이 있는 위치를 담은 배열 rocks, 제거할 바위의 수 n이 매개변수로 주어질 때, 바위를 n개 제거한 뒤 각 지점 사이의 거리의 최솟값 중에 가장 큰 값을 return 하도록 solution 함수를 작성해주세요.

제한사항

• 도착지점까지의 거리 distance는 1 이상 1,000,000,000 이하입니다.
• 바위는 1개 이상 50,000개 이하가 있습니다.
• n 은 1 이상 바위의 개수 이하입니다.

입출력 예

distance	rocks	n	return
25	[2, 14, 11, 21, 17]	2	4

풀이

mid: 최솟값

remove: 제거한 돌의 수

이분 탐색을 이용하여 푸는문제

자세한 내용은 코드 주석에 있음

코드

import java.util.*;
class Solution {
    public int solution(int distance, int[] rocks, int n) {
        Arrays.sort(rocks);
        int answer = 0;
        int start = 0;
        int end = distance;
        int prev, mid, remove;
        while (start <= end){
            remove = 0;
            mid = (start + end) / 2;
            prev = 0;
            for (int i = 0; i < rocks.length; i++){
                if (rocks[i] - prev < mid) //최솟값인 mid보다 더 작은 값이 존재하면 안되기 때문에 바위 제거
                    remove++;
                else
                    prev = rocks[i];
            }
            if(distance - rocks[rocks.length-1] < mid) // 마지막 돌과 도착 지점의 거리가 mid보다 작으면 안되기 때문에 바위 제거
                remove++;
            if(remove <= n){
                answer = mid;
                start = mid + 1;
            }
            else{
                end = mid - 1;
            }
        }
        return answer;
    }
}

 

후기

start랑 end는 쉽게 찾았는데 mid를 뭐로 둬야 될지 헷갈렸다