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

소수점 아래 숫자가 계속되지 않고 유한개인 소수를 유한소수라고 합니다.

분수를 소수로 고칠 때 유한소수로 나타낼 수 있는 분수인지 판별하려고 합니다.

유한소수가 되기 위한 분수의 조건은 다음과 같습니다.

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

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

 

제한사항

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

import java.util.ArrayList;

class Solution {
    public int solution(int a, int b) {
        
        int answer = 0;
        int gcd = gcd(a, b);
        Boolean flag = true;
        
        ArrayList<Integer> arr = new ArrayList<>();
        
        a /= gcd;
        b /= gcd;
        
        //분모 소인수분해
        for(int i=2; i<=b; i++) {
            if(b % i == 0) {
                while(b % i == 0) {
                    b /= i;
                }
                arr.add(i);
            }
        }

        //분모의 소인수가 2,5로만 이루어져있으면 true 아니면 false
        for(int i=0; i<arr.size(); i++) {
            if(!(arr.get(i) == 2 || arr.get(i) == 5)) {
                flag = false;
            }
        }

        if(flag) {
            answer = 1;
        }else answer = 2;
        
        return answer;
    }
    
    //최대공약수 구하는 함수
    private static int gcd(int a, int b) {
        if (b == 0) {
            return a;
        }
        return gcd(b, a % b);
    }
}