본문 바로가기

Programming/코딩테스트 준비

[프로그래머스] 124 나라의 숫자 (C++)

728x90

programmers.co.kr/learn/courses/30/lessons/12899

 

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

 

programmers.co.kr

문제 설명

124 나라가 있습니다. 124 나라에서는 10진법이 아닌 다음과 같은 자신들만의 규칙으로 수를 표현합니다.

  1. 124 나라에는 자연수만 존재합니다.
  2. 124 나라에는 모든 수를 표현할 때 1, 2, 4만 사용합니다.

예를 들어서 124 나라에서 사용하는 숫자는 다음과 같이 변환됩니다.

10진법 124 나라 10진법 124 나라
1 1 6 14
2 2 7 21
3 4 8 22
4 11 9 24
5 12 10 41

자연수 n이 매개변수로 주어질 때, n을 124 나라에서 사용하는 숫자로 바꾼 값을 return 하도록 solution 함수를 완성해 주세요.

 

제한사항

  • n은 500,000,000이하의 자연수 입니다.

입출력 예

n result
1 1
2 2
3 4
4 11

 

풀이

10진법->3진법 변환 문제다.

다만, 0,1,2로 이루어진 3진법이 아니라 1,2,4로 이루어진 3진법이다. 

그리고 원래 3진법에서는 0을 0으로 나타낼 수 있지만, 문제에서는 1부터 1로 시작하므로 3진법에 대응할 수 있도록 나머지를 계산하거나 몫을 구할때 n-1을 기준으로 계산했다. (이걸 어떻게 말로 잘 설명할 수가 없지만 코드를 보면 무슨 말인지 대충 이해가 될 것이다.)

str124라는 벡터에 124 진법에 맞게 한자리씩 추가를 해주면 가장 오른쪽 자리 수 부터 들어가게 되므로, 완성된 str124벡터의 역순이 정답이다.

 

코드

#include <string>
#include <vector>
using namespace std;

string solution(int n) {
    string answer = "";
    char to124[3] = {'1','2','4'};
    vector<char> str124;
    // 3진법으로 바꾸면서 1,2,4에 대응시켜 str124에 추가
    int i=0;
    while(1){
        str124.push_back(to124[(n+2)%3]);
        n = (n-1)/3;
        if(n==0) break;
    }
    // 역순으로 answer에 추가
    while(!str124.empty()){
        answer += str124.back();
        str124.pop_back();
    }
    return answer;
}