[c++] 프로그래머스 - 124 나라의 숫자(Level 2)

문제

코딩테스트 연습 - 124 나라의 숫자

 

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

string solution(int n) {
    string answer = "";
    
    while(n){
        bool flag = false;
        if(n%3==0){
            answer+='4';
            flag = true;
        }
        if(n%3==1) answer+='1';
        if(n%3==2) answer+='2';
        n/=3;
        if(flag) n-=1;
    }
    reverse(answer.begin(),answer.end());
    return answer;
}

 

풀이

• N % 3 == 0 일 때 4
• N % 3 == 1 일때 1
• N % 3 == 2 일때 2

위 규칙으로 124나라의 숫자를 생성할 수 있습니다. 다만 N이 3의 약수일 때 문제가 생기는데 3의 약수일 때는 N-1을 해줌으로써 문제를 해결할 수 있습니다. 

그 이유는 기본적으로 n진수는 0~n-1까지의 숫자인데, 124 나라에서는 0이 없이 4가 존재하기 때문에 발생합니다.