코딩테스트 -- 마법의 엘리베이터 - (프로그래머스 / C++)

프로그래머스

 

문제!

---

마법의 세계에 사는 민수는 아주 높은 탑에 살고 있습니다. 탑이 너무 높아서 걸어 다니기 힘든 민수는 마법의 엘리베이터를 만들었습니다. 마법의 엘리베이터의 버튼은 특별합니다. 마법의 엘리베이터에는 -1, +1, -10, +10, -100, +100 등과 같이 절댓값이 10c (c ≥ 0 인 정수) 형태인 정수들이 적힌 버튼이 있습니다. 마법의 엘리베이터의 버튼을 누르면 현재 층 수에 버튼에 적혀 있는 값을 더한 층으로 이동하게 됩니다. 단, 엘리베이터가 위치해 있는 층과 버튼의 값을 더한 결과가 0보다 작으면 엘리베이터는 움직이지 않습니다. 민수의 세계에서는 0층이 가장 아래층이며 엘리베이터는 현재 민수가 있는 층에 있습니다.

마법의 엘리베이터를 움직이기 위해서 버튼 한 번당 마법의 돌 한 개를 사용하게 됩니다.예를 들어, 16층에 있는 민수가 0층으로 가려면 -1이 적힌 버튼을 6번, -10이 적힌 버튼을 1번 눌러 마법의 돌 7개를 소모하여 0층으로 갈 수 있습니다. 하지만, +1이 적힌 버튼을 4번, -10이 적힌 버튼 2번을 누르면 마법의 돌 6개를 소모하여 0층으로 갈 수 있습니다.

마법의 돌을 아끼기 위해 민수는 항상 최소한의 버튼을 눌러서 이동하려고 합니다. 민수가 어떤 층에서 엘리베이터를 타고 0층으로 내려가는데 필요한 마법의 돌의 최소 개수를 알고 싶습니다. 민수와 마법의 엘리베이터가 있는 층을 나타내는 정수 storey 가 주어졌을 때, 0층으로 가기 위해 필요한 마법의 돌의 최소값을 return 하도록 solution 함수를 완성하세요.

 

---

 

간단히 요약하면!  ?층에 있는 민수를 0층으로 이동시키는 데에 마법의 돌을 최소로 사용하게 하는 갯수를 구하는 문제입니다.

 

 

만약 16층에 있다면 1번을 4번 -10번을 2번 눌러 0 층으로 이동가능합니다.

 

 

여기서 좀더 생각해 본다면 ???층에 민수가 있을때 모든 자릿수에 대해 0이 될때 까지 버튼을 눌러 이동하면 쉽게 해결할수 있어 보입니다.

 

 

그럼 코드 시작!

---

 

먼저 자릿수를 나누어 새각하기 위해 10으로 계속 나누어 num 벡터에 저장해줍시다.

vector<int>num;

int solution(int storey) {    //0을 만들어내는것이 가장 최소
    int answer = 0;
    
    while(storey){
        num.push_back(storey%10);
        storey/=10;
    }
    num.push_back(0);
    
	//~~~
    
    return answer;
}

 

 

다음은 num에 저장되어있는 수들을 순서대로(1자리 10자리 100자리...)

5를 기준으로 +해줄지 -해줄지를 판단해주어야합니다.

  for(int i=0;i<num.size();i++){
        if(num[i]>5){
            answer+=(10-num[i]);
            num[i+1]+=1;
        }
        else if(num[i]==5){
            //0이나 10에 가까워야함
            if(cal(num[i+1])>cal(num[i+1]+1)){//넘기는게 더 좋네?
                answer+=(10-num[i]);
                num[i+1]+=1;
            }
            else{
                answer+=num[i];
            }
        }
        else{
            answer+=num[i];
        }
    }

 

여기서 num[i]=5 일때는 + 를 해줘야할지 - 를 할지를 판단해야 하는데 

 

저는 cal 함수를 만들어

+해서 다음 자리에도 +1 해주는 경우와 

-를 하는 경우를 비교해주어 

 

넘겨줄지말지를 결정 해 주었습니다.

int cal(int memo){
    if(memo>5){
        return 10-memo;
    }
    else{
        return memo;
    }
}

 

 

여기까지 하면 모든 경우처리가 끝났습니다.

 

 

끝!

#include <string>
#include <vector>
#include <iostream>

using namespace std;

vector<int>num;


int cal(int memo){
    if(memo>5){
        return 10-memo;
    }
    else{
        return memo;
    }
}

int solution(int storey) {    //0을 만들어내는것이 가장 최소
    int answer = 0;
    
    while(storey){
        num.push_back(storey%10);
        storey/=10;
    }
    num.push_back(0);
    
    for(int i=0;i<num.size();i++){
        if(num[i]>5){
            answer+=(10-num[i]);
            num[i+1]+=1;
        }
        else if(num[i]==5){
            //0이나 10에 가까워야함
            if(cal(num[i+1])>cal(num[i+1]+1)){//넘기는게 더 좋네?
                answer+=(10-num[i]);
                num[i+1]+=1;
            }
            else{
                answer+=num[i];
            }
        }
        else{
            answer+=num[i];
        }
    }

    
    
    return answer;
}

 

 

 

 

틀린점이 있다면 댓 달아주세요!