[프로그래머스] 📐징검다리

😀문제 설명

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

---

🤣입출력 예 설명

문제에 나온 예와 같습니다.

출처

※ 공지 - 2020년 2월 17일 테스트케이스가 추가되었습니다.
※ 공지 - 2023년 5월 15일 테스트케이스가 추가되었습니다. 기존에 제출한 코드가 통과하지 못할 수도 있습니다.

---

😎나의 풀이

해당 문제는 이분탐색이 필요한 문제이다. 필요한 풀이는 주석으로 남겨두었음

function solution(distance, rocks, n) {
    // 1. rocks array에 출발점과 도착점을 추가하고 오름차순으로 정렬
    rocks.push(0);
    rocks.push(distance);
    rocks.sort((a,b)=>a-b);

    // 2. 각 지점 사이의 거리를 계산하여 배열에 저장
    let gaps = [];
    for(let i=1; i<rocks.length; i++) {
        gaps.push(rocks[i] - rocks[i-1]);
    }

    // 3. 이진 탐색을 위한 초기 설정
    let left = 0;
    let right = distance;

   while(left <= right){
        // mid는 바위 사이의 거리 제한
        const mid = Math.floor((left + right)/2); 
        let minGap = Infinity;
        let sumGap = 0;
        let removeCount = 0;

       for(let gap of gaps){
            sumGap += gap;      // 다음 gap과 합침 (바위 제거)
            if(sumGap < mid){ // 현재 gap으로 mid보다 작으면 
                removeCount++;       // 제거된 바위 count
            } else {                // 현재 gap으로 mid보다 크면 
                minGap = Math.min(minGap,sumGap);   //min값 업데이트 
                sumGap=0;           //sum 초기화 (새로운 구간 시작)
            }
       }

       if(removeCount > n){         //제거해야하는 바위수 보다 많으면 
           right=mid-1;             //(mid가 너무 커서 많은 바위가 제거된 경우)
       }else{                       //(mid가 너무 작아서 적게 제거된 경우)
           left=mid+1;
       }
   }

   return right;
}