[Programmers] n진수 게임

문제 바로가기

def solution(n, t, m, p):
    chars = {num : chr(num+55) for num in range(10,17)}
    for i in range(10):
        chars[i] = str(i)
    # n 진법 수 구하기
    def parse(num):
        res = []
        while num > 0:
            num, k = divmod(num, n)
            res.append(chars[k])
        return reversed(res)
    ls = ['0']
    i = 1
    while len(ls) < m*t + 1:
        ls.extend(parse(i))
        i+=1
    answer = [ls[i] for i in range(m*t) if i % m == p-1]
    return ''.join(answer)

ls 에 순서대로 때려박고 가져오기