[본 문제는 정확성과 효율성 테스트 각각 점수가 있는 문제입니다.]
카카오 초등학교의 "니니즈 친구들"이 "라이언" 선생님과 함께 가을 소풍을 가는 중에 징검다리가 있는 개울을 만나서 건너편으로 건너려고 합니다. "라이언" 선생님은 "니니즈 친구들"이 무사히 징검다리를 건널 수 있도록 다음과 같이 규칙을 만들었습니다.
"니니즈 친구들"은 개울의 왼쪽에 있으며, 개울의 오른쪽 건너편에 도착해야 징검다리를 건넌 것으로 인정합니다.
"니니즈 친구들"은 한 번에 한 명씩 징검다리를 건너야 하며, 한 친구가 징검다리를 모두 건넌 후에 그 다음 친구가 건너기 시작합니다.
디딤돌에 적힌 숫자가 순서대로 담긴 배열 stones와 한 번에 건너 뛸 수 있는 디딤돌의 최대 칸수 k가 매개 변수로 주어질 때, 최대 몇 명까지 징검다리를 건널 수 있는지 return 하도록 solution함수를 완성해주세요.
[입출력 예]
stones | k | result |
---|---|---|
[2, 4, 5, 3, 2, 1, 4, 2, 5, 1] | 3 | 3 |
[제한사항]
첫번 째 풀이
const solution = (stones, k) => {
let answer = 0;
while(true){
let count = 0;
for(let i = 0 ; i < stones.length ; i++){
if(stones[i] !== 0){
stones[i]--;
count = 0;
}else{
count++;
}
if(count >= k){
break;
}
}
if(count >= k){
return answer;
}else{
answer ++;
}
}
}
우선 시간복잡도 생각하지 않고 맞춰보자는 식으로 코드를 짰습니다. 한 사람이 징검다리를 건너고 있음을 for
문으로 나타냈고, 다음 사람을 위해 한번 더 반복되어야 하니 while
문으로 감싸는 형태가 되었습니다.
해당 시간복잡도는 O(N²)이며, 정확성은 다 맞추었지만 효율성은 0점인 풀이입니다.
두번 째 풀이
const solution = (stones, k) => {
return BSearch(stones, k, 1, Math.max.apply(null, stones));
}
const BSearch = (list, k, min, max) => {
if(min === max){
return min;
}
let mid = Math.round((min + max) / 2);
let count = 0;
for(let i = 0 ; i < list.length ; i++){
if(count === k)break;
let value = list[i]-(mid-1);
value <= 0 ? count++ : count = 0;
}
if(count === k){
return BSearch(list, k, min, mid-1);
}else{
return BSearch(list, k, mid, max);
}
}
재귀함수와 이진탐색을 이용해 시간을 상당히 줄였습니다.
정확성 테스트에서는 모두 성공을 보여줬고, 효율성 테스트에서도 시간초과 대신 걸린시간과 메모리 용량을 보여주네요. 아무래도 답이 틀렸나봅니다.
세번 째 풀이
const solution = (stones, k) => {
return BSearch(stones, k, 1, 200000000);
}
const BSearch = (list, k, min, max) => {
if(min === max){
return min;
}
let mid = Math.round((min + max) / 2);
let count = 0;
for(let i = 0 ; i < list.length ; i++){
if(count === k)break;
let value = list[i]-(mid-1);
value <= 0 ? count++ : count = 0;
}
if(count === k){
return BSearch(list, k, min, mid-1);
}else{
return BSearch(list, k, mid, max);
}
}
위의 코드와 똑같이만 범위를 제한사항에서 나온 최대 디딤돌 숫자인 200,000,000으로 정해주고 시작했습니다. 그랬더니..
짜잔~ 통과되었습니다. ㅎㅎ