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

문제 설명

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

기약분수로 나타내었을 때, 분모의 소인수가 2와 5만 존재해야 합니다.
두 정수 a와 b가 매개변수로 주어질 때, a/b가 유한소수이면 1을, 무한소수라면 2를 return하도록 solution 함수를 완성해주세요.

---

제한사항

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

---

입출력 예

---

 

이 문제는 (1) 기약분수를 구한 다음, (2) 분모에 대해서 소인수분해를 진행하여 2와 5로만 이루어졌는지 확인하면 된다.

두 단계 모두 지난 입문 과정에서 풀었던 최대공약수 구하기와 소인수분해 두 문제를 활용하면 되는 간단한 문제라고 볼 수 있다. 

 

하지만, 기억이 나지 않아서 ... 지난 글들을 참고하였다. 

 

먼저 최대공약수 구하는 코드는 아래와 같다.

def GCD(a, b):
	while b:
    	a, b = b, a%b
    return a

 

그 다음, 소인수분해하는 코드는 아래와 같다.

def factor(n):
    lst, d = [], 2
    while n > 1:
         f n%d==0:
            lst.append(d)
            n = n//d
        else:
            d += 1
    return lst

 

 

 

최대공약수로 나눈 분모에 대해, lst 속에 모둔 원소들이 2 또는 5로만 이루어졌는지 if not all 을 활용하여 확인하면 된다. 

if not all(x in [2,5] for x in answer):
        return 2
    else:
        return 1

 

최종 코드는 아래와 같다.

def solution(a, b):
    def GCD(a, b):
        while b:
            a, b = b, a%b
        return a
    
    def factor(n):
        lst, d = [], 2
        while n > 1:
            if n%d==0:
                lst.append(d)
                n = n//d
            else:
                d += 1
        return lst
    
    n = GCD(a, b)
    answer = factor(b//n)
    if not all(x in [2,5] for x in answer):
        return 2
    else:
        return 1

 

끝 !