Kotlin / 콜라츠 추측

문제 설명

1937년 Collatz란 사람에 의해 제기된 이 추측은, 주어진 수가 1이 될 때까지 다음 작업을 반복하면, 모든 수를 1로 만들 수 있다는 추측입니다. 작업은 다음과 같습니다.

1-1. 입력된 수가 짝수라면 2로 나눕니다. 
1-2. 입력된 수가 홀수라면 3을 곱하고 1을 더합니다. 
2. 결과로 나온 수에 같은 작업을 1이 될 때까지 반복합니다. 

예를 들어, 주어진 수가 6이라면 6 → 3 → 10 → 5 → 16 → 8 → 4 → 2 → 1 이 되어 총 8번 만에 1이 됩니다. 위 작업을 몇 번이나 반복해야 하는지 반환하는 함수, solution을 완성해 주세요. 단, 주어진 수가 1인 경우에는 0을, 작업을 500번 반복할 때까지 1이 되지 않는다면 –1을 반환해 주세요.

제한 사항

• 입력된 수, num은 1 이상 8,000,000 미만인 정수입니다.

입출력 예nresult

6	8
16	4
626331	-1

입출력 예 설명

입출력 예 #1
문제의 설명과 같습니다.

입출력 예 #2
16 → 8 → 4 → 2 → 1 이 되어 총 4번 만에 1이 됩니다.

입출력 예 #3
626331은 500번을 시도해도 1이 되지 못하므로 -1을 리턴해야 합니다.

 

풀이

재귀함수를 만들어 조건이 될때까지 반복하여 함수를 호출했다.

 

탈출 조건 : 

주워진 수가 1 -> 0

500번 반복시 -> -1

반복해서 결과가 1이 되면 -> 반복한 수

 

반복 조건 : 

짝수 -> 수/2

홀수 -> 수*3 + 1

 

코드

fun solution1(num: Int): Int {
    return collatz(num.toLong(),0)
}
fun collatz(num:Long,answer:Int):Int{
    if(answer+1 == 500) return -1
    if(answer == 0 && num == 1L) return 0
    if(num == 1L) return answer
    return if(num%2 == 0L){
        collatz(num/2, answer+1 )
    }else{
        collatz(num*3+1, answer+1 )
    }
}

 

처음엔 Long을 생각 못하고, Int로 인자를 넘겨주었다.

같은 로직인데도 마지막 테스트 케이스에서 488이 나왔다.

626331(케이스 3)인 경우, 계산 중에 int 자료형의 범위를 초과하게 된다고 한다.

따라서 Int를 Long으로 바꿔서 돌려주니 테스트를 통과했다.

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/12943