[프로그래머스] 최대공약수와 최소공배수 - python

최대공약수와 최소공배수

문제 설명

두 수를 입력받아 두 수의 최대공약수와 최소공배수를 반환하는 함수, solution을 완성해 보세요. 배열의 맨 앞에 최대공약수, 그다음 최소공배수를 넣어 반환하면 됩니다. 예를 들어 두 수 3, 12의 최대공약수는 3, 최소공배수는 12이므로 solution(3, 12)는 [3, 12]를 반환해야 합니다.

제한사항

• 두 수는 1이상 1000000이하의 자연수입니다.

풀이

최소공배수 = a * b / 최대 공약수

def solution(n, m):
    a1 = set([x for x in range(1, (n//2 + 1)) if n % x == 0])
    a2 = set([x for x in range(1, (m//2 + 1)) if m % x == 0])
    a1.add(n)
    a2.add(m)
    a = max(a1 & a2)
    return [a, n * m // a]

다른 사람 풀이

유클리드 호제법을 이용한 풀이

나머지가 0이 될 때까지 큰 수에서 작은 수로 MOD 연산을 반복한다.

def solution(a, b):
    c, d = max(a, b), min(a, b)
    t = 1
    while t > 0:
        t = c % d
        c, d = d, t
    answer = [c, int(a*b/c)]
​
    return answer