프로그래머스 - 최대공약수와 최소공배수 / Python

문제: https://programmers.co.kr/learn/courses/30/lessons/12940

코딩테스트 연습 - 최대공약수와 최소공배수

문제 설명

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

제한 사항

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

입출력 예nmreturn

3	12	[3, 12]
2	5	[1, 10]

입출력 예 설명

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

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

 

---

정답

def solution(n):
    answer = n ** 0.5
    return (answer + 1) ** 2 if answer == int(answer) else -1

 

---

풀이

 파이썬에서는 math.gcd()를 이용하여 최대공약수를 구할 수 있고, math.lcm()을 이용하여 최소공배수를 구할 수 있습니다. 그러나 프로그래머스에서는 math.lcm()을 지원하지 않으므로 다음 공식을 이용하여 최소공배수를 구하면 됩니다.

 

최소공배수 = n * m / 최대공약수