[프로그래머스 lv1] 최대공약수와 최소공배수 (파이썬)

문제 설명

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

제한 사항

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

입출력 예

입출력 예

---

나의 풀이

from math import gcd
# python 3.9 부터는 math.lcm 사용 가능

def solution(n, m):
    return [gcd(n, m), n * m // gcd(n, m)]

최대공약수와 최소공배수의 개념을 알면 쉽게 풀 수 있는 문제.

테스트 결과

추가 (라이브러리 사용 x)

def solution(n, m):
    gcd = lambda a,b : b if not a%b else gcd(b, a%b)
    lcm = lambda a,b : a*b//gcd(a,b)
    return [gcd(n, m), lcm(n, m)]