프로그래머스 lv0 유한소수 판별하기 (Kotlin)

문제 설명

소수점 아래 숫자가 계속되지 않고 유한개인 소수를 유한소수라고 합니다. 분수를 소수로 고칠 때 유한소수로 나타낼 수 있는 분수인지 판별하려고 합니다. 유한소수가 되기 위한 분수의 조건은 다음과 같습니다.

• 기약분수로 나타내었을 때, 분모의 소인수가 2와 5만 존재해야 합니다.

두 정수 a와 b가 매개변수로 주어질 때, a/b가 유한소수이면 1을, 무한소수라면 2를 return하도록 solution 함수를 완성해주세요.

 

제한사항

• a, b는 정수
• 0 < a ≤ 1,000
• 0 < b ≤ 1,000

 

Hint

• 분자와 분모의 최대공약수로 약분하면 기약분수를 만들 수 있습니다.
• 정수도 유한소수로 분류합니다.

 

풀이

class Solution {
    fun solution(a: Int, b: Int): Int {
        var calB = b / gcd(a, b)
        while (calB != 1) {
            calB /= if (calB % 2 == 0) {
                2
            } else if (calB % 5 == 0) {
                5
            } else {
                return 2
            }
        }
        return 1
    }
    
    //유클리드 호제법(최대공약수 구하기)
    private fun gcd(a: Int, b: Int): Int {
        return if (b == 0) {
            a
        } else {
            gcd(b, a % b)
        }
    }
}

 

유클리드 호제법을 이용해 최대공약수를 구한다.

분모(b)를 최대공약수로 나누고 2와 5로 나눈 나머지 값이 0이 될 때까지 반복문을 실행한다.

b가 1이라면 유한소수이므로(분모의 소인수가 2와 5만 존재) 1을 리턴하고,
b가 그 외의 값이라면 무한소수이므로(분모에 2와 5를 제외한 소수가 포함) 2를 리턴한다.