코테 공부/프로그래머스

[Python] N진수 게임 (10진수를 N진수로 변환하는 함수)

prefer_all 2022. 10. 19. 10:58
문제

튜브가 활동하는 코딩 동아리에서는 전통적으로 해오는 게임이 있다. 이 게임은 여러 사람이 둥글게 앉아서 숫자를 하나씩 차례대로 말하는 게임인데, 규칙은 다음과 같다.

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

 

 

입출력 예제
n t m p result
2 4 2 1 "0111"
16 16 2 1 "02468ACE11111111"
16 16 2 2 "13579BDF01234567"
풀이

아이디어: 10진수를 n진수로 변환한 값을 t*m개 나열한 후 튜브 차례만 뽑아낸다.

                   10진수 게임일 때 t*m개의 10진수 값을 변환한 값이 필요함. 2진수, 16진수로 바꾸면 10진수 값을 t*m개나 변환할 필요 없음에도.

 

10진수 -> n진수로의 변환은 재귀함수를 이용한 함수 사용

매개변수로 두개의 숫자를 입력받아 몫과 나머지를 튜플로 반환한다.   
divmod(10, 3) 이라 할때, 10을 3으로 나누고 (= 10 / 3) 그 몫인 3과 나머지인 1을 튜플 형식으로 (3, 1)로 반환 하는 함수
def convert(i, n):
    temp = "0123456789ABCDEF"
    q, r = divmod(i, n) # i//n, i%n
    if q == 0:
        return temp[r]
    else:
        return convert(q, n) + temp[r]
    
def solution(n, t, m, p):
    # n진법, 미리 구할 숫자의 개수 t, 게임에 참가하는 인원 m, 튜브의 순서 p
    # 튜브가 말해야 하는 숫자 t개를 공백 없이 차례대로 출력
    
    # 10진수를 n진수로 변환할 값을 "t*m개" 나열하기  (10진수가 아니라면 t*m보다 적게 필요할 수 있음)  
    # p-1번째, (p-1)+m번째, (p-1)+2m번째 ... 값을 answer에 concat
    
    numbers = '' # 모든 인원이 말하는 숫자들
    for i in range(t*m): # 십진수 i를 n진수로 변환
        numbers += convert(i,n)
    
    tube = '' # 튜브가 말하는 숫자
    idx = p-1
    while len(tube) < t:
        tube += numbers[idx]
        idx += m
    return tube

 

튜브가 말할 숫자 뽑아내는 거 아래 코드도 된다

 while len(ans) < t:
        ans += arr[p-1]
        p += m

 

tip!

- idx는 순서보다 하나 작아야 한다는 점까지 고려해서 꼼꼼히 처리

- string으로 return해야하므로 list에 append하는 식이 아닌 string에 concat하는 식으로 코드 작성

 

출처

다른 풀이

위와 거의 동일하나 튜브 차례만 추출할 때 for문에 step을 넣어주는 방식

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

def solution(n, t, m, p):
    answer = ""
    temp = []
    
    # 튜브가 t개의 숫자를 말해야하므로 전체 m명이 t번 돌아야함
    for i in range(t*m):
        base_num = num_to_base(i, n) # i(번째)를 n진법으로 변환
        for c in base_num:
            temp.append(c)
    
    # 튜브 차례 숫자만 추출
    for i in range(p-1, t*m, m):
        answer += temp[i]

            
    return answer

📝

n진수 → 10진수

  • int(string, base)
print(int('111',2))	#7
print(int('222',3))	#26
print(int('333',4))	#63
print(int('444',5))	#124
print(int('555',6))	#215
print(int('FFF',16)) 	#4095

 

10진수 → 2,8,16진수

  • 2, 8, 16진수는 bin(), oct(), hex() 함수를 지원
print(bin(10))	#0b1010
print(oct(10))	#0o12
print(hex(10))	#0xa
  • 0b는 2진수, 0o는 8진수, 0x는 16진수를 의미
  • 앞의 진법 표시를 지울려면 [2:]를 하면 됨

 

10진수 → n진수

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]