[프로그래머스] n진수 게임 (python)

코딩테스트 연습 - [3차] n진수 게임

 

👩🏻‍💻 코드

def convert(n, base):
    T = "0123456789ABCDEF"
    q, r = divmod(n, base)
    if q == 0:
        return T[r]
    else:
        return convert(q, base) + T[r]


def solution(n, t, m, p):
    MAX = t * m + 1
    num_list = []

    for i in range(MAX):
        if len(convert(i, n)) > 1:
            num_list += list(convert(i, n))
        else:
            num_list += convert(i, n)

    return ''.join(num_list[i * m + (p - 1)] for i in range(t))

 

📝 정리

convert 함수는 https://m.blog.naver.com/PostView.naver?isHttpsRedirect=true&blogId=teenager4282&logNo=220952327548 를 참고하였다. 파이썬은 divmod 함수가 있어서 몫과 나머지를 한 번에 구할 수 있다.