[프로그래머스] N개의 최소공배수 - python

N개의 최소공배수

문제설명

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

제한사항

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

풀이

a * b = gcd * lcm

math 모듈의 gcd 사용해도 되지만 그냥 유클리드호제법으로 구현해봤다.

앞에서부터 두 개씩 최소 공배수를 구한다

def gcd(a, b):
    if a % b == 0:
        return b 
    return gcd(b, a % b)

def lcm(a, b):
    return (a * b) // gcd(a, b)

def solution(arr):
    answer = arr[0]
    for n in arr:
        answer = lcm(n , answer)
    return answer