[프로그래머스 LV.0] 유한소수 판별하기

유한소수 판별하기

문제 설명

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

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

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

제한 사항

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

입출력 예

a	b	result
7	20	1
11	22	1
12	21	2

입출력 예 설명

입출력 예 #1

• 분수 7/20은 기약분수입니다. 분모 20의 소인수가 2, 5 이기 때문에 유한소수입니다. 따라서 1을 return 합니다.

입출력 예 #2

• 분수 11/22는 기약분수로 나타내면 1/2입니다. 분모 2는 소인수가 2 뿐이기 때문에 유한소수입니다. 따라서 1을 return 합니다.

입출력 예 #3

• 분수 12/21는 기약분수로 나타내면 4/7입니다. 분모 7은 소인수가 7 이므로 무한소수입니다. 따라서 2를 return 합니다.

 

Hint

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

---

제출

import Foundation

func gcd(_ a: Int, _ b: Int) -> Int {
    return b == 0 ? a : gcd(b, a%b)
}

func solution(_ a:Int, _ b:Int) -> Int {
    var b = b/gcd(a,b)
    
    if b == 1 {return 1}
    
    while (b%2 == 0 || b%5 == 0) {
        b = b%2 == 0 ? b/2 : b/5
        if b == 1 {return 1}
    }
    return 2
}

분모와 분자의 최대공약수를 구한 후 약분해 기약 분수를 만든다.
분모가 1이라면 정수, 유한소수이기 때문에 1을 반환한다.
분모가 2 또는 5로 나누어지지 않을 때까지 반복한다. 분모가 1이 되면 1을 아니라면 2를 반환한다.