[프로그래머스] 레벨4 (level4) 징검다리

2020년 08월 27일, 21:40

징검다리

문제 설명

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

문제풀이

  1. rocks들을 정렬해서 서로 거리를 구할 수 있게 한다.
  2. 이분 탐색을 이용해서 푼다.
  3. rocks를 돌면서, mid값보다 작은 수가 있을 경우에는 바위를 제거.
  4. mid값보다 크다면, 바위의 위치를 바꿈.
  5. count에 따라서 left,right를 바꿔줌.
/*
n만큼 제거 해야함.
제거 시 거리의 최솟값을 구할거임.
그 거리들의 최솟값 중 최댓값을 구함.

mid가 최솟값이라고 가정.
mid보다 작은 애가 있다면 돌을 지움.
계속 진행. count가 n보다 크면 right를 줄이고, count가 n보다 작으면 left를 늘림.
*/

function solution(distance, rocks, n) {
  var answer = 0;
  rocks.sort((a, b) => a - b);
  let left = 0;
  let right = distance;
  let maxValue = 0;

  while (left <= right) {
    let mid = parseInt((left + right) / 2);
    let prev = 0;
    let min = Infinity;
    let count = 0;

    for (let i = 0; i < rocks.length; i++) {
      if (count > n) break;
      let d = rocks[i] - prev;
      if (mid > d) {
        count++;
      } else {
        min = min > d ? d : min;
        prev = rocks[i];
      }
      d = distance - prev;
      min = min > d ? d : min;
    }
    if (count <= n) {
      left = mid + 1;
      maxValue = Math.max(maxValue, min);
    } else {
      right = mid - 1;
    }
  }
  answer = maxValue;
  return answer;
}