Programmers 징검다리 java 풀이

문제 설명

출발지점부터 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 이상 바위의 개수 이하입니다.

 

나의 풀이

 이번 문제는 이분탐색으로 해결하기 위해 상당히 애를 먹은 문제.. 이해하는데도 시간이 좀 걸렸고 다른 사람의 풀이들을 찾아보며 이해하고 재작성해본 것인데도 뭔가 내꺼가 된 느낌이 아직 안든다...

 풀이는 코드를 작성하면서 생각이 정리되는데로 주석으로 넣었다

 이분탐색 개념이 아니라 이분탐색으로 문제에 적용하고 풀어가는 방법에 대해 특공이 필요하다 느꼈다

 

코드

import java.util.*;

class Solution {
    public int solution(int distance, int[] rocks, int n) {
        int answer = 0;
        Arrays.sort(rocks);
        int pl = 0;
        int pr = distance;
        int pc, removeCount, lastRock;
        
        // 임의의 거리를 pc라고 놓고, 바위들간 거리의 최솟값으로 pc가 되기 위해 돌을 제거한다.
        // 만약 pc가 최솟값이 유지 안되는 돌이 있다면 해당 돌을 제거해버릴 것이다.
        // 몇 개의 돌을 제거하면 될지 이진검색으로 검색한다.
        // n개만큼의 돌을 제거해서 pc가 최솟값이 된다면 그 pc를 answer에 저장한다.
        // 얻은 pc 값중에 제일 큰값을 asnwer에 저장하도록 반복하며 Math.max로 answer값을 갱신한다.
        
        while(pl<=pr) {
            // 처음 제거한 돌의 수는 0에서 시작,
            removeCount = 0;
            // 처음 돌의 위치는 0부터 시작(출발점)
            lastRock = 0;
            // 최솟값을 pc라고 임의로 설정한다.
            pc = (pl+pr)/2;
            
            //pc가 최솟값이 되도록 for문으로 모든 돌들의 위치를 확인해 돌을 제거하거나 그대로 둔다.
            for(int i=0; i<rocks.length; i++) {
                // 만약 마지막에 있던 돌(혹은 시작점) - 현재돌 만큼의 거리가
                // pc보다 작으면 pc가 돌사이 거리 중 최솟값이 될 수 없기때문에
                // rocks[i] 현재 돌을 제거한다고 본다.
                if(rocks[i]-lastRock < pc) removeCount++;
                
                // 마지막에 있던 돌 - 현재돌 만큼의 거리가 pc보다 멀다면
                // 현재 돌을 제거하지 않아도 pc가 돌사이 거리 중 최솟값으로 유지된다.
                // 그리고 다음 돌과의 현재돌 사이에 거리를 측정하기 위해 현재돌을 마지막 돌에 대입한다.
                else lastRock = rocks[i];
            }
            
            //종착점 - 마지막에 있던 돌 사이의 거리가 pc보다 작다면
            //마지막 돌을 제거해야만 pc가 최솟값으로 유지 될 수 있다.
            if(distance - lastRock < pc) removeCount++;
            
            if(removeCount <=n) {
                // pc가 최솟값이 되기 우해서 제거한 돌의 수가 지정된 n보다 작다면
                // pc를 이전에 구했던 answer과 비교해서 큰 값을 저장한다.
                answer = Math.max(answer, pc);
                // 조건을 만족하는 다른 pc값이 없는지 다시 재조사 한다.
                pl = pc+1;
            } else {
                //지정한 n보다 더 많은 수의 돌을 제거해야만 pc가 최솟값이 된다면,
                //이 조건은 성립하지 않기 때문에 pc값을 갱신한다.
                pr = pc - 1;
            }
            
        }
        
        return answer;
    }
}