[프로그래머스] Level1 - 최대공약수와 최소공배수 (Python)

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

제한 사항

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

입출력 예

n	m	return
3	12	[3, 12]
2	5	[1, 10]

입출력 예 설명

입출력 예 #1
위의 설명과 같습니다.

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

내 문제 풀이

• 최대공약수는 math library에 있기 때문에 import math의 gcd 사용
• 최소공배수도 math library에 있는데 (lcm, 3.9 버전 이상) 해보니 돌아가질 않아서 직접 구현 함.
  -> n과 m 중에 큰 수를 작은수로 나눴을 때 나머지가 0이면 큰 수를 return
  -> 나머지가 0이 아닌 경우엔 n과 m을 곱한 수 최대공약수로 나눈 값

import math

def solution(n, m):

    return [math.gcd(n,m), max(n,m) if max(n,m) % min(n,m) == 0 else n*m/math.gcd(n,m)]

(ex. 20과 50의 경우 최대공약수는 10이고 최소공배수를 구하기 위해 내 로직을 적용하면 50 % 20 != 0 이기 때문에 두 수를 곱하는데 (1000) 1000은 두 수의 최소공배수가 아니다. 여기엔 두 수의 최대공약수인 10이 2번 곱해져있기 때문에 최대공약수인 10으로 나눠야 100이라는 정확한 최소공배수를 얻을 수 있다)

아쉬운 점

• 직접 최대공약수를 구하는 로직을 구현하지 않고 python 내장 함수를 썼다는 점