[알고리즘] n진수 게임 (프로그래머스 2018 Kakao Blind Recruitment)

코딩테스트 연습 - [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	result
2		2	1	"0111"
16	16	2	1	"02468ACE11111111"
16	16	2	2	"13579BDF01234567"

입력 형식

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

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

출력 형식

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

해설 보러가기

💡나의 문제 풀이

문제 해결 아이디어

• 0부터 어떠한 수까지 n진수로 표현하기 위해서는 각 정수에 대한 나눗셈을 진행해 몫과 나머지를 얻는 과정을 반복해야 한다. 몫이 해당 정수로 나뉘어질 수 있다면, 이 과정을 반복하여 얻어지는 나머지들을 거꾸로 나열하면 된다.
• n진수에 대하여, 미리 구할 숫자의 갯수 t에 게임에 참가하는 인원 m을 곱한 숫자의 길이만큼 구함
• 튜브의 순서 (p-1) 번째부터 시작하여, (m-1 번째 문자들만 골라 만든 길이 t만큼의 문자열을 리턴함)
  • 0부터 시작하여, 숫자를 더해가며 리스트에 넣음
    • 리스트의 길이가 (t * m ) 이 될 때까지 반복
    • 이후에 (m -1) 번째 인덱스값들만 골라 문자열로 이어붙여 리턴

고려사항

• 나머지가 10~15의 숫자인 경우, 즉 10진수에서 16진수로 변환하여 얻어지는 나머지들에 대해서는, ABCDEF로 표현해야 함.

작성 코드

• 나머지에 대해 10~15 수에 대해서는 사전을 이용하였다.
• 정확성 57.7점으로 답안 코드들과 동일한 아이디어로 구현했지만, 정확히 각 케이스들에 대하여 왜 실패가 발생했는지는 원인을 발견하지 못했다. 😢

# n진수, 말해야 하는 t개, m명, p번째 순서
def solution(n, t, m, p):
    arr = []
    dic = {10:'A', 11:'B', 12:'C', 13:'D', 14:'E', 15:'F'}
    for i in range(t*m):
        l = [] # 나머지 넣기
        j=i
        while True:
            r = j%n 
            if r in dic: # 나머지 추가
                l.append(dic[r]) # 10보다 크다면 알파벳 추가 
            else:
                l.append(str(r))
            j//=n # 남은 몫
            if j<n:
                l.append(str(j))
                break

        l.reverse()
        if l[0]=='0':
            del l[0]
        arr.append(''.join(l))
    arr = ''.join(arr)

    answer=''
    for i in range(p-1, t*m, m):
        answer+=arr[i]
    return answer[:t]

def test():
    print(solution(2,4,2,1)=="0111")
    print(solution(16,16,2,1)=="02468ACE11111111")
    print(solution(16,16,2,2)=="13579BDF01234567")

test()

def convert(num, base):
    tmp = '0123456789ABCDEF'
    q, r = divmod(num, base)
    if q==0: # 몫이 0인 경우
        return tmp[r]
    else:
        # q를 base로 변환
        return convert(q,base)+tmp[r]

# n진수, 말해야 하는 t개, m명, p번째 순서
def solution(n, t, m, p):
    answer = ''
    s = ''
    for i in range(t*m):
        s += str(convert(i,n))
    # 순서인 부분만 추출
    for i in range(p-1, t*m, m):
        answer += s[i]
    return answer

답안 코드

• 동일한 아이디어와 로직을 기반으로 작성된 답안 코드였다. num을 base로 나눌 때 얻게 되는 몫과 나머지를 리턴하는 divmod(num, base) 를 사용해 진수 변환을 간단하게 구현했다.

def convert(num, base):
    temp = "0123456789ABCDEF"
    q, r = divmod(num, base)

    if q == 0:
        return temp[r]
    else:
        # q를 base로 변환
        # 즉, n진수의 다음 자리를 구함
        return convert(q, base) + temp[r]

def solution(n, t, m, p):
    answer = ''
    test = ''

    for i in range(m*t):
        test += str(convert(i, n))

    while len(answer) < t:
        answer += test[p-1]
        p += m

    return answer

리뷰

• n진수 → 10진수 : int(string, base)
• 10진수 → 2,8,16 진수 : bin(n) , oct(n) , hex(n)
• divmod(num, base) 함수 : 몫과 나머지를 튜플 형식으로 리턴