[프로그래머스 lv2] N개의 최소공배수 (파이썬)

문제 설명

두 수의 최소공배수(Least Common Multiple)란 입력된 두 수의 배수 중 공통이 되는 가장 작은 숫자를 의미합니다. 예를 들어 2와 7의 최소공배수는 14가 됩니다. 정의를 확장해서, n개의 수의 최소공배수는 n 개의 수들의 배수 중 공통이 되는 가장 작은 숫자가 됩니다. n개의 숫자를 담은 배열 arr이 입력되었을 때 이 수들의 최소공배수를 반환하는 함수, solution을 완성해 주세요.

제한 사항

• arr은 길이 1이상, 15이하인 배열입니다.
• arr의 원소는 100 이하인 자연수입니다.

입출력 예

입출력 예

 

---

나의 풀이

from math import gcd

def solution(arr):
    for i in range(1, len(arr)):
        arr[i] = arr[i-1] * arr[i] // gcd(arr[i-1], arr[i])
    return arr[-1]

파이썬에는 math 라이브러리에 gcd 메소드(최대공약수)가 있다.

추가로 파이썬 3.9x 부터는 lcm 메소드(최소공배수)도 생겼다.

 

프로그래머스 컴파일 옵션

하지만 아쉽게도 프로그래머스에서 지원하는 파이썬 버전은 3.8 버전이기 때문에 lcm 메소드를 사용할 수 없다.

 

그래도 gcd를 사용하여 최소공배수를 구하는 것은 어렵지 않다.

 

방법은 간단하다.

숫자 2개씩 차례대로 최소공배수를 구하면 된다.

 

a, b의 최소 공배수를 구하는 식은

a x b ÷ (a, b의 최대공약수) 이다.

 

나는 배열을 순회하며 각 자리에 계산 결과를 저장했는데

이 부분은 파이썬의 reduce 함수를 사용했어도 좋을 것 같다.

 

 

참고로 math.gcd를 사용할 수 없을 때 최대공약수를 구하는 코드는 다음과 같다.

# 일반적인 함수
def gcd(a, b):
	if not a % b:
    	    return b
    else:
    	return gcd(b, a%b)

# 람다식으로 표현한 경우
gcd = lambda a, b: a if not b else gcd(b, a%b)

 

유클리드 호제법을 사용한 방법이다.

 

위 두개의 코드의 원리는 같다.

 

테스트 결과