문제

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

숫자를 0부터 시작해서 차례대로 말한다. 첫 번째 사람은 0, 두 번째 사람은 1, … 열 번째 사람은 9를 말한다.
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

풀이

아이디어

• 파이썬에는 10진수를 N진수로 바꿔주는 메서드가 따로 없기 때문에 직접 convert 함수를 만들어주었다.
  • 이때 10부터 15까지는 'A', 'B', ... 'F'를 써줘야하므로 Dictionary Comprehension을 사용하여 매핑시켜주었다.
• 나머지는 문제에서 시키는 대로 하면 되는데, 튜브의 순서 p번째마다 문자를 채택하는 방법에 나는 문제가 있었다. 현재까지 진행된 인덱스가 i라고 하면 (i+1) % m == p 일때마다 문자를 채택해주었었는데, m과 p가 같으면 (ex. m = 2, p = 2) 의도했던 p번째 숫자가 아니고 0번째가 되어서 채택되지 않았다.
  • 따라서 m==p이면 사전에 p를 0으로 만들어서 코드를 통과시켰는데, 다른 사람의 코드를 보니 이는 리스트 슬라이싱에서의 step을 이용하면 됐었다..! (하..ㅋㅋ 파이썬 진짜 사기다)
  • 또, 리스트 슬라이싱으로 t번 까지만 구하면 된다.
• 다른 사람의 코드에서 그냥 DIGITS = list('0123456789ABCDEF')로 한방에 배열을 만들어준 아이디어가 있었는데, 이것도 되게 좋은 것 같다. 문자열에다가 list 함수를 씌우면 각각의 문자로 이루어진 배열이 된다는 사실을 이용한 것이다.

코드

#10진법 수 num을 k진법의 수로 문자열 형식으로 바꿔주는 함수
def convert(num, k):
    table = {i:chr(i+55) for i in range(10, 16)}
    ret = ''
    
    if num == 0:
        return '0'
    
    while num > 0:
        if num % k >= 10:
            ret = table[num % k] + ret
        else:
            ret = str(num % k) + ret
        num = num // k
    
    return ret

def solution(n, t, m, p):
    temp = ''
    count = 0
    while len(temp) < m*t:
        temp += convert(count, n)
        count += 1
    
    answer = ''
    
    if m == p:
        p = 0
    for i in range(len(temp)):
        if len(answer) < t and (i+1) % m == p:
            answer = answer + temp[i]
    
    return answer

리스트 슬라이싱을 활용한 간단한 마지막 코드

#10진법 수 num을 k진법의 수로 문자열 형식으로 바꿔주는 함수
def convert(num, k):
    table = {i:chr(i+55) for i in range(10, 16)}
    ret = ''
    
    if num == 0:
        return '0'
    
    while num > 0:
        if num % k >= 10:
            ret = table[num % k] + ret
        else:
            ret = str(num % k) + ret
        num = num // k
    
    return ret

def solution(n, t, m, p):
    temp = ''
    count = 0
    while len(temp) < m*t:
        temp += convert(count, n)
        count += 1
    
    # if m == p:
    #     p = 0
    # for i in range(len(temp)):
    #     if len(answer) < t and (i+1) % m == p:
    #         answer = answer + temp[i]
    
    answer = temp[(p-1)::m][:t]
    
    return answer