[Algorithm] 프로그래머스 - 콜라츠 수열 만들기 (Kotlin)

 

while 문의 조건을 설정하는 데서 실수해서 시간을 잡아먹은 문제다. 그 외에는 문제의 조건대로 주먹구구식으로 풀었다.

 

class Solution {
    fun solution(n: Int): IntArray {
        var aNum = n
        val list = mutableListOf<Int>()
        list.add(n)

        while (aNum != 1) {
            if (aNum % 2 == 0) {
                aNum /= 2
                list.add(aNum)
            } else {
                aNum = (3 * aNum + 1)
                list.add(aNum)
            }
        }

        return list.toIntArray()
    }
}