190201 이분탐색 (1) 징검다리

2019-02-01

이분탐색 (1) 징검다리

문제 설명

출발지점부터 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

풀이 과정
  • 답은 베껴왔지만 아직 완전히 이해하지는 못함 ㅠㅠ
  • 내 추측으론 이분탐색을 통해 각 지점 사이 거리의 최솟값중 큰 값을 리턴하는듯?
  • hits가 n보다 작거나 같은 경우는 바위 n개 제거했을때 최솟값중 최댓값이 a가 되는 것이 불가능하므로, a를 c로 옮겨 탐색하는것같다.
import java.util.*;

class Solution {
    public int solution(int distance, int[] rocks, int n) {
 		Arrays.sort(rocks);
        int a=0;
        int b=distance;
        while(a<b){
            int c=(a+b+1)/2;
            int p=0;
            int hits=0;
            for(int i=0;i<rocks.length;i++){
                if(rocks[i]-p<c){
                    ++hits;
                }else{
                    p=rocks[i];
                }
            }
            
            if(hits>n){
                b=c-1;
            }else{
                a=c;
            }
        }
        return a;
    }
}