1110번: 더하기 사이클

문제 출처 : https://www.acmicpc.net/problem/1110

1110번: 더하기 사이클

언어 : Kotlin

 

문제 설명 :

0보다 크거나 같고, 99보다 작거나 같은 정수가 주어질 때 다음과 같은 연산을 할 수 있다. 먼저 주어진 수가 10보다 작다면 앞에 0을 붙여 두 자리 수로 만들고, 각 자리의 숫자를 더한다. 그 다음, 주어진 수의 가장 오른쪽 자리 수와 앞에서 구한 합의 가장 오른쪽 자리 수를 이어 붙이면 새로운 수를 만들 수 있다. 다음 예를 보자.

26부터 시작한다. 2+6 = 8이다. 새로운 수는 68이다. 6+8 = 14이다. 새로운 수는 84이다. 8+4 = 12이다. 새로운 수는 42이다. 4+2 = 6이다. 새로운 수는 26이다.

위의 예는 4번만에 원래 수로 돌아올 수 있다. 따라서 26의 사이클의 길이는 4이다.

N이 주어졌을 때, N의 사이클의 길이를 구하는 프로그램을 작성하시오.

• 첫째 줄에 N이 주어진다. N은 0보다 크거나 같고, 99보다 작거나 같은 정수이다.
• 첫째 줄에 N의 사이클 길이를 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
26	4
55	3
1	60
0	1
71	12

풀이 : 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

var cnt = 0

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val n = br.readLine().toInt()
    cycle(n, n)

    bw.write("$cnt")
    bw.flush()
    bw.close()
}

fun cycle(n: Int, x: Int) {
    val temp = x.toString().chunked(1)
    val rightNum = temp.last()
    val sumNum = temp.sumOf { it.toInt() }.toString().chunked(1).last()
    val newNum = if (x >= 10) { ("$rightNum$sumNum").toInt() } else ("$rightNum$x").toInt()
    cnt++
    
    if (newNum != n) cycle(n, newNum)
}