[Python] 백준 1110번

1110번 더하기 사이클

0보다 크거나 같고, 99보다 작거나 같은 정수가 주어질 때 다음과 같은 연산을 할 수 있다. 먼저 주어진 수가 10보다 작다면 앞에 0을 붙여 두 자리 수로 만들고, 각 자리의 숫자를 더한다. 그 다음, 주어진 수의 가장 오른쪽 자리 수와 앞에서 구한 합의 가장 오른쪽 자리 수를 이어 붙이면 새로운 수를 만들 수 있다. 26부터 시작한다. 2+6 = 8이다. 새로운 수는 68이다. 6+8 = 14이다. 새로운 수는 84이다. 8+4 = 12이다. 새로운 수는 42이다. 4+2 = 6이다. 새로운 수는 26이다. N이 주어졌을 때, N의 사이클의 길이를 구하는 프로그램을 작성하시오.

Input)
26

Output)
4

문제

최초 입력값(n)이 10보다 작을 경우 앞에 0을 붙이는 로직을 while문 안에서 수행하다 보니 루프를 1번 진행한 후 10보다 작은 수(n을 new_n값으로 치환할 때)가 될 경우 0을 한번 더 붙여버리는 문제가 발생하였다.

문제해결

최초 입력값(n)이 10보다 작을 경우에만 앞에 0을 붙이도록 while문 바깥쪽에서 수행하도록 변경하였다. 코드가 더러워졌다는 단점이 있다..

n = input()
new_n,count = 0,0
first_n = n
if int(n) < 10: # 최초 입력값이 10보다 작을 경우 앞에 0을 붙임.
    n = '0' + n
while True:
    if int(n) == 0:
        count = 1
        break
    rs = int(n[0])+int(n[1]) # 두수의 합
    if rs >= 10:
        new_n = n[1] + str(rs)[1] # 두수의 합이 10보다 클 경우 우측자리수로 저장.
    else:
        new_n = n[1] + str(rs) 
    n = new_n
    count = count+1
    if int(new_n) == int(first_n):
        break
print(count)