[Python] [백준] 6378번 디지털 루트

6378번: 디지털 루트

문제

양의 정수 N의 디지털 루트를 구하려면 N을 이루고 있는 모든 자리수를 더해야 한다. 이때, 더한 값이 한 자리 숫자라면, 그 수가 N의 디지털 루트가 된다. 두 자리 이상 숫자인 경우에는 다시 그 수를 이루고 있는 모든 자리수를 더해야 하며, 한 자리 숫자가 될 때 까지 반복한다.

24의 디지털 루트를 구해보자. 2+4=6이다. 6은 한 자리 숫자이기 때문에, 24의 디지털 루트는 6이 된다. 39의 경우에는 3+9=12이기 때문에, 한 번 더 더해야 한다. 따라서, 1+2=3이 디지털 루트가 된다.

양의 정수 N이 주어졌을 때, 그 수의 디지털 루트를 구하는 프로그램을 작성하시오.

입력

한 줄에 하나씩 양의 정수가 주어진다. 입력의 마지막 줄은 0으로 나타낸다.  수는 최대 1000자리이다.

출력

입력으로 주어진 양의 정수의 디지털 루트를 한 줄에 하나씩 주어진 순서대로 출력한다.

예제

예제 입력1	예제 출력1
24 39 0	6 3

코드

def dig_root(n):
    s = sum(list(map(int, n)))
    if s < 10:
        return s
    else:
        return dig_root(str(s))

while 1:
    n = input()
    if n == '0':
        break
    else:
        print(dig_root(n))

 

설명

문제의 39처럼 각 자리의 합이 두 자리 이상일 수 있기 때문에 재귀 함수로 풀었다.

각 자리의 합은 list로 한 자리씩 쪼갠 후 합했다.