[프로그래머스] N진수 게임 [2018 KAKAO BLIND RECRUITMENT]

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

코딩테스트 연습 - [3차] n진수 게임

 

N진수 게임

튜브가 활동하는 코딩 동아리에서는 전통적으로 해오는 게임이 있다. 이 게임은 여러 사람이 둥글게 앉아서 숫자를 하나씩 차례대로 말하는 게임인데, 규칙은 다음과 같다.

1. 숫자를 0부터 시작해서 차례대로 말한다. 첫 번째 사람은 0, 두 번째 사람은 1, … 열 번째 사람은 9를 말한다.
2. 10 이상의 숫자부터는 한 자리씩 끊어서 말한다. 즉 열한 번째 사람은 10의 첫 자리인 1, 열두 번째 사람은 둘째 자리인 0을 말한다.

이렇게 게임을 진행할 경우,
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 0, 1, 1, 1, 2, 1, 3, 1, 4, …
순으로 숫자를 말하면 된다.

한편 코딩 동아리 일원들은 컴퓨터를 다루는 사람답게 이진수로 이 게임을 진행하기도 하는데, 이 경우에는
0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 0, 1, 1, 1, …
순으로 숫자를 말하면 된다.

이진수로 진행하는 게임에 익숙해져 질려가던 사람들은 좀 더 난이도를 높이기 위해 이진법에서 십육진법까지 모든 진법으로 게임을 진행해보기로 했다. 숫자 게임이 익숙하지 않은 튜브는 게임에 져서 벌칙을 받는 굴욕을 피하기 위해, 자신이 말해야 하는 숫자를 스마트폰에 미리 출력해주는 프로그램을 만들려고 한다. 튜브의 프로그램을 구현하라.

입력 형식

진법 n, 미리 구할 숫자의 갯수 t, 게임에 참가하는 인원 m, 튜브의 순서 p 가 주어진다.

• 2 ≦ n ≦ 16
• 0 ＜ t ≦ 1000
• 2 ≦ m ≦ 100
• 1 ≦ p ≦ m

출력 형식

튜브가 말해야 하는 숫자 t개를 공백 없이 차례대로 나타낸 문자열. 단, 10~15는 각각 대문자 A~F로 출력한다.

입출력 예제

n	t	m	p	result
2	4	2	1	"0111"
16	16	2	1	"02468ACE11111111"
16	16	2	2	"13579BDF01234567"

풀이

1. 0부터 차례로 n진법으로 변환한 문자열을 만든다.
2. 앞에서부터 제거하며 튜브의 차례를 찾는다.
3. 차례가 되면 result에 이어붙인다.
4. 문자열이 비면 숫자를 1 올려 t 번 반복.

import string

tmp = string.digits + string.ascii_uppercase
def convert(num, base) :
    q, r = divmod(num, base)
    if q == 0 :
        return tmp[r] 
    else :
        return convert(q, base) + tmp[r]

def solution(n, t, m, p):
    i = 0; cnt = 0
    answer = ''; result = ''
    while cnt < t * m + p - 1:
        if answer == '':
            answer = convert(i, n)
            i += 1
            
        if cnt % m == p - 1:
            result += answer[0]
            
        answer = answer[1:]
        cnt += 1
    
    return result

진법 변환이 익숙치 않아 재귀함수를 이용하였습니다. 경험 상 코딩 테스트에서 함수를 두 개 이상 사용하는 경우는 거의 없으며, 코드가 길어지는 경우도 적고 메모리 제한 때문에 재귀를 이용하는 상황도 드뭅니다. 마침 이 문제는 제한 사항이 제시되어 있지 않기 때문에 답변만 제대로 출력한다면 정답으로 처리되지만, 공부하는 입장에서 보다 효율적인 코드를 알아봅시자.

다른 사람의 풀이

다른 사람은 어떻게 풀었는지 확인하기 위해 "다른 사람의 풀이" 메뉴에서 가장 짧고, 상단에 위치한 풀이를 가져왔습니다. 보기가 너무 어렵네요. 적어도 재귀 함수는 없다는 점에서 더 높은 점수를 줄 것 같습니다.

big = ["A","B","C","D","E","F"]
def solution(n, t, m, p):
    a="0"
    i=0
    #for i in range(t*m):
    while True:
        if len(a)>=t*m:
            break
        b=""
        j=i
        while (j):
            if j%n>9:
                b=big[j%n-10]+b
            else:
                b=str(j%n)+b
            j=j//n
        a=a+b
        i=i+1
    answer = a[p-1::m][:t]
    return answer

보기 쉽도록 변수에 제대로 된 이름을 붙이고 정리해봅시다. 재귀만 while 문으로 변경되었을 뿐 형태는 비슷해지네요. 함수 형태로 분리하는 게 변수 이름에도 좋고, mutation 문제도 해결할 수 있어 보입니다.

characterType = '0123456789ABCDEF'

def convert(num, n):
    converted = ''
    while num:
        num, mod = divmod(num, n)
        converted = characterType[mod] + converted
    return converted

def solution(n, t, m, p):
    answer = '0'; num = 0
    while len(answer) < t * m:
        answer = answer + convert(num, n)
        num += 1
        
    return answer[p-1::m][:t]

카카오 해설

tech.kakao.com/2017/11/14/kakao-blind-recruitment-round-3/

카카오 신입 공채 3차 코딩 테스트 문제 해설