문제
문제 설명
카카오 초등학교의 "니니즈 친구들"이 "라이언" 선생님과 함께 가을 소풍을 가는 중에 징검다리가 있는 개울을 만나서 건너편으로 건너려고 합니다. "라이언" 선생님은 "니니즈 친구들"이 무사히 징검다리를 건널 수 있도록 다음과 같이 규칙을 만들었습니다.
- 징검다리는 일렬로 놓여 있고 각 징검다리의 디딤돌에는 모두 숫자가 적혀 있으며 디딤돌의 숫자- 는 한 번 밟을 때마다 1씩 줄어듭니다.
- 디딤돌의 숫자가 0이 되면 더 이상 밟을 수 없으며 이때는 그 다음 디딤돌로 한번에 여러 칸을 건너 뛸 수 있습니다.
- 단, 다음으로 밟을 수 있는 디딤돌이 여러 개인 경우 무조건 가장 가까운 디딤돌로만 건너뛸 수 있습니다.
"니니즈 친구들"은 개울의 왼쪽에 있으며, 개울의 오른쪽 건너편에 도착해야 징검다리를 건넌 것으로 인정합니다.
"니니즈 친구들"은 한 번에 한 명씩 징검다리를 건너야 하며, 한 친구가 징검다리를 모두 건넌 후에 그 다음 친구가 건너기 시작합니다.
디딤돌에 적힌 숫자가 순서대로 담긴 배열 stones와 한 번에 건너뛸 수 있는 디딤돌의 최대 칸수 k가 매개변수로 주어질 때, 최대 몇 명까지 징검다리를 건널 수 있는지 return 하도록 solution 함수를 완성해주세요.
제한 사항
- 징검다리를 건너야 하는 니니즈 친구들의 수는 무제한 이라고 간주합니다.
- stones 배열의 크기는 1 이상 200,000 이하입니다.
- stones 배열 각 원소들의 값은 1 이상 200,000,000 이하인 자연수입니다.
- k는 1 이상 stones의 길이 이하인 자연수입니다.
입출력 예
stones k result [2, 4, 5, 3, 2, 1, 4, 2, 5, 1] 3 3
문제 풀이
처음엔 당연하게도 while문을 통해서 징검다리의 값들을 하나씩 줄여가면서 for문을 이용해서 0이 연속으로 k번 나오면 종료시키도록 구현을 하려했지만 효율성 0점이 나왔고.. (당연함)
결국 문제 풀이 힌트를 찾아보던 중.. 이진 탐색 이용하면 된다는 말을 봤지만?
'아니 stones 배열은 정렬도 못하는데 어떻게 이진 탐색을 쓴다는거지??' 했는데 알고보니 stones 를 탐색하는 것이 아닌! 문제에서 주어진 stones 배열 각 원소들의 값은 1 이상 200,000,000 이하인 자연수입니다. 이 조건을 이용해서 징검다리를 건널 수 있는 사람 수를 이진 탐색해서 찾는거였다!!!
for(let i=0; i<arr.length; i++){
if(check === k) break;
let temp = arr[i] - (mid-1);
temp <= 0 ? check++ : check = 0;
}여기서 check는 징검다리 값이 연속으로 0인 경우를 count 해주는 변수이다.
temp값이 0보다 작거나 같다면 check를 증가시키고, 아닐시 check를 다시 0으로 초기화 해주는 방식으로 연속으로 0이 나타나는 경우를 확인할 수 있다!
만약 check === k 일 경우에는 해당 범위에선 답을 찾을 수 없으므로 break 해준다.
정답 코드
function solution(stones, k) {
let min = 1;
let max = 200000000;
const binarySearch = (arr, min, max) => {
if(min === max) return min;
let check = 0;
let mid = Math.round((min+max)/2);
for(let i=0; i<arr.length; i++){
if(check === k) break;
let temp = arr[i] - (mid-1);
temp <= 0 ? check++ : check = 0;
}
if(check === k){
return binarySearch(arr, min, mid-1);
}else{
return binarySearch(arr, mid, max);
}
}
return binarySearch(stones, min, max);
}
Frontend Developer