Programmers / [3차] n진수 게임 / Python 파이썬

 

*문제 출처는 프로그래머스에 있습니다.

문제 제목: [3차] n진수 게임 (2단계)

문제 사이트: https://school.programmers.co.kr/learn/courses/30/lessons/17687

프로그래머스

문제 설명

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로 출력한다.

입출력 예제

ntmpresult

2	4	2	1	"0111"
16	16	2	1	"02468ACE11111111"
16	16	2	2	"13579BDF01234567"

---

나의 풀이 (처음 풀이) - 53점

def solution(n, t, m, p):
    # 진법 n, 미리 구할 숫자의 갯수 t, 게임에 참가하는 인원 m, 튜브의 순서 p

    answer = ''

    nums = ['0']
    cnt = 0

    # 진법을 하나씩 늘여서 저장
    while len(nums) <= m * t:
        cnt += 1
        tmp = jinsu(n, cnt)
        for i in tmp:
            nums.append(i)

    # 해당 순서에 반환

    answer += nums[p - 1]
    idx = p - 1
    while len(answer) != t:
        idx += m
        answer += nums[idx]

    return answer


def jinsu(n, l):

    rev_base = ''
    if 2 <= n <= 10:
        while l > 0:
            l, mod = divmod(l, n)
            rev_base += str(mod)
        return rev_base[::-1]

    else:  # 16진법
        return hex(l)[2:].capitalize()

틀린 이유: 10 이상의 수는 A,B,C,D,E,F로 취급해야한다는 것을 빼먹음 -> 10진법이상은 16진법으로 바로 계산해서 틀렸다

 

def solution(n, t, m, p):
    # 진법 n, 미리 구할 숫자의 갯수 t, 게임에 참가하는 인원 m, 튜브의 순서 p

    answer = ''

    nums = ['0']
    cnt = 0

    # 진법을 하나씩 늘여서 저장
    while len(nums) <= m * t:
        cnt += 1
        tmp = jinsu(n, cnt)
        for i in tmp:
            nums.append(i)

    # 해당 순서에 반환

    answer += nums[p - 1]
    idx = p - 1
    while len(answer) != t:
        idx += m
        answer += nums[idx]

    return answer


def jinsu(n, l):

    rev_base = ''
    alpha = ['A', 'B', 'C', 'D', 'E', 'F']
    while l > 0:
        l, mod = divmod(l, n)
        if mod >= 10:
            rev_base += alpha[mod - 10]
        else:
            rev_base += str(mod)
    return rev_base[::-1]

모범답안 1

def solution(n, t, m, p):
    answer = ''
    alpha = ['A', 'B', 'C', 'D', 'E', 'F']
    num = 0
    result = '0'
    
    while len(answer) < t:
        temp = ''
        a = num
        
        while a > 0:
            a, b = divmod(a, n)
            if b >= 10:
                temp = alpha[b-10] + temp
            else:
                temp = str(b) + temp
        
        result += temp
        answer = ''
        
        for i in range(p-1, len(result), m):
            answer += result[i]
            
        num += 1
        
    return answer[:t]

 

모범답안 2

DIGITS = list('0123456789ABCDEF')

def n2base(n, base):
    if n == 0:
        return DIGITS[0]

    # 각 자리수에 해당하는 문자열을 담을 리스트
    digits = []
    while n > 0:
        # 제일 마지막 자리의 숫자 구하기. 예를 들어 1658이면 '8'
        digits.append(DIGITS[n % base])
        # 제일 마지막 자리 제거. 예를 들어 1658이면 165로
        n = int(n // base)

    # 뒤집어서 반환. 예를 들어 '8561'이면 '1658'
    return ''.join(digits[::-1])


def solution(n, t, m, p):
    digits = []
    turn = 0
    while len(digits) < t * m:
        digits += list(n2base(turn, n))
        turn += 1
    return ''.join(digits[p-1::m][:t])

진법을 변환해주는 함수 'n2base'를 정의하여, 그 수들을 이어 붙인 리스트 'digits'의 길이가 't * m' 이상일 때까지 반복하며 구한 후, 'digits'를 'p-1'부터 'm'씩 건너 뛴 값을 't'개 만큼 join()함수를 이용해 하나의 문자열로 만들어주었다. 진법을 변환해주는 함수에서는 값이 0일 때 'DIGITS[0]'을 리턴하는 것과, 그냥 이어 붙였던 것을 뒤집이서 문자열로 반환하기 위해 'join(digits[::-1])'을 해준 것을 볼 수 있었다.

 

 

다른 풀이 출처: https://ziho-world.tistory.com/5

---

※ 알아야 할 것

- 10진법 다음은 16진법이라고 생각하지말기..^^