[Python] n진수 게임

https://school.programmers.co.kr/learn/courses/30/lessons/17687

프로그래머스

# n 진수 게임
# n 진수로 숫자 변환 함수
def n_radix(n, num):
    a = num
    num_ = ''
    num2alpha = {10:'A', 11:'B', 12:'C', 13:'D', 14:'E', 15:'F'}
    while a > 0:
        rest = a % n
        a = a//n
        if rest < 10:
            num_ += str(rest)
        else:
            num_ += num2alpha[rest]
    num_ = num_[::-1]
    return num_

def solution(n, t, m, p):
    
    answer = ''
    for i in range(t*m):
        if i==0:
            answer += '0'
        else:
            answer += n_radix(n, i)
            
    answer = answer[p-1:t*m:m]
    return answer