[Cos Pro 1급, Python] 5차 6번 : p진법 to q진법

문제

p 진법으로 표현한 수란, 각 자리를 0부터 p-1의 숫자로만 나타낸 수를 의미합니다. p 진법으로 표현한 자연수 두개를 더한 결과를 q 진법으로 표현하려 합니다.

예를 들어, 3진법 수 112001과 12010을 더한 결과를 8진법으로 나타내면 1005입니다.

solution 함수의 매개변수로 p 진법 자연수를 담은 문자열 s1, s2와 두 수를 나타내는 진법의 기수 p, 두 수의 덧셈 결과를 표현할 진법의 기수 q가 매개변수로 주어집니다. p진법으로 표현된 두 수를 더한 결과를 q 진법으로 나타낸 값을 return 하도록 solution 함수를 완성해주세요.

 

 

 

 

 

 

 

 

 

 

코드

def solution(s1, s2, p, q):
	answer = ''
	rev_base = ''
	result = 0
	
	result += int(s1, p) + int(s2, p)
	
	while result > 0:
		result, mod = divmod(result, q)
		rev_base += str(mod)
	
	answer = rev_base[::-1]
	return answer
    
    
    
    

s1 = "112001"
s2 = "12010"
p = 3
q = 8
ret = solution(s1, s2, p, q)

print("solution 함수의 반환 값은", ret, "입니다.")