Programmers with Python [Level1] - 최대공약수와 최소공배수

문제 설명

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

• 제한 사항

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

• 입출력 예시
  

• 입출력 예시 설명

  입출력 예 #1
  위의 설명과 같습니다.
  입출력 예 #2
  자연수 2와 5의 최대공약수는 1, 최소공배수는 10이므로 [1, 10]을 리턴해야 합니다.

코드

# https://school.programmers.co.kr/learn/courses/30/lessons/12940
# programmers, level1 - 최대공약수와 최소공배수, python3
import math
def solution(n, m):
    x = math.gcd(n, m)
    y = n*m/x
    return [x,y]

# or

def solution(n, m):
    x, y = max(n, m), min(n, m)
    t = 1
    while (t > 0):
        t = x % y
        x, y = y, t
    answer = [x, n*m//x]
    return answer

결과