[프로그래머스] 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 이상 바위의 개수 이하입니다.

✔ 접근방법

이 문제도, 도착지점까지의 거리 값이 10억인 것을 보고 이분탐색을 이용해야한다는 것을 추측할 수 있다.

여기서 이분탐색 mid값으로 사용하는 값은 바위들 사이의 거리값이다.

각 바위의 위치를 탐색하면서, 이전에 위치한 바위와, 현재 바위의 거리값이 mid보다 작은 경우엔 바위를 제거하는 식으로 문제를 해결한다.

1. 바위를 제거하면 이전 바위값(before)이 변경되지 않는다는 점
2. 바위를 제거하는 값이 n보다 작아야 한다는 점
3. n보다 더 많이 바위를 제거하는 경우엔 mid값을 증가시켜야 한다는 점

위 사항만 잘 고려하여 코드를 작성한다면 문제를 해결할 수 있다고 생각한다.

✔ 코드

import java.util.*;
class Solution {
    public int solution(int distance, int[] rocks, int n) {
        int answer = 0;
        int left = 0;
        int right = distance;
        
        Arrays.sort(rocks);
        while(left <= right){
            int mid = (left + right)/2;
            int count = 0;
            
            int before = 0;
            int current = rocks[0];
            for(int i = 0; i < rocks.length; i++){
                current = rocks[i];
                if(current - before < mid){
                    count++; //돌제거
                }
                else{
                    before = current;
                }
                //제거할 수 있는 바위의 수를 넘음 = mid값이 큼
                if(count > n){                	
                    break;
                }
                
                
            }
            
            //마지막 위치까지 추가 계산
            if(count <= n){
                current = distance;
                if(current - before < mid){
                    count++; //돌제거
                }
            }
            
            if(count > n){
                right = mid - 1;
            }
            else{
                answer = mid;
                left = mid + 1;
            }
            
        }
        return answer;
    }
}

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