[프로그래머스] 마법의 엘리베이터 JS

문제링크

풀이 방법

처음에는 한 번 bfs로 해봤는데 수가 커서 그런지 2개 빼고 다 시간초과가 걸리더라구요 그냥 규칙을 찾아서 카운트 해줬습니다

고려해야 하는 점은 수가 5일 때였습니다.

 

수가 5일 때

1. 다음 자리수가 5 이상이면 +5를 더해서 다음 자리수를 +1 해주고
2. 다음 자리수가 5 미만이면 -5 해줬습니다

5보다 큰 경우에는 무조건 다음 자리 값을 올려줬습니다

만약 마지막 자리수인 경우에는 1을 푸쉬해서 자리수를 올려줬습니다

 

 

function solution(storey) {
  
    let cnt=0;
    const storey_arr = String(storey).split('').map(Number).reverse();
    

   for(let i=0;i<storey_arr.length;i++){
       const val = storey_arr[i];
       const hasNext = i+1<storey_arr.length;
       //5이하일 때는 바로 -버튼
       if(val<5){
           cnt+=val;
           continue;
       }
       
       //5일 때
       if(val===5){
           //앞 수가 5이상이면 +1해줘야 함
           if(hasNext&&storey_arr[i+1]>=5){
               storey_arr[i+1]++;
               cnt+=5;
               continue;
           }
           //앞 수가 5보다 작으면 그냥 5만 더함
           cnt+=5;
           continue;
       }
       
       //5보다 클 때 무조건 다음 자리 값 올려줘야 함
       if(val>5){
           cnt+=10-val;
           hasNext?storey_arr[i+1]++:storey_arr.push(1);
       }
   }
    return cnt;
}