문제 설명

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

기약분수로 나타내었을 때, 분모의 소인수가 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

나의 코드

function solution(a, b) {
    let i = 2;
    let arr = []
    while(i<=b){
        if(b%i === 0){
            b = b/i
            if(i !== 2 && i !==5) arr.push(i)
        }else{
          i++
        }
    }
    if(arr === []) return 1
    b = arr.reduce((a,b)=>a*b,1)
    return a%b === 0 ? 1 : 2 
}

풀이

기약분수의 분모가 2 또는 5로 이루어진 경우에만 1을 반환하고 그외에는 2를 반환한다

분모 b의 2이상인 약수 중 2, 5가 아닌 수의 집합을 배열 arr이라고 할 때

let i = 2;
let arr = []
while(i<=b){
    if(b%i === 0){
        b = b/i
        if(i !== 2 && i !==5) arr.push(i)
    }else{
      	i++
    }
}

arr의 모든 요소의 곱을 b라고 할 때
a에 b를 나눈 나머지가 0이면 1을 반환하고 그렇지 않으면 2를 반환한다

b = arr.reduce((a,b)=>a*b,1)
return a%b === 0 ? 1 : 2

TIL

처음에 b가 2 또는 5의 배수인 경우 arr이 빈배열[]이 되어 반환 조건을 아래와 같이 작성했는데

if(arr === []) return 1

다시 생각해보니 어쨌든 reduce()에서 초기값을 1로 설정했기 때문에 위 조건식이 없어도 잘 동작한다
reduce는 배열의 합으로만 사용했었는데 그외에도 곱으로도 사용할 수 있음을 알았다.

b = arr.reduce((a,b)=>a*b,1)