N개의 최소공배수(python)
참고 자료
문제 설명
두 수의 최소공배수(Least Common Multiple)란 입력된 두 수의 배수 중 공통이 되는 가장 작은 숫자를 의미합니다.
예를 들어 2와 7의 최소공배수는 14가 됩니다.
정의를 확장해서, n개의 수의 최소공배수는 n 개의 수들의 배수 중 공통이 되는 가장 작은 숫자가 됩니다.
n개의 숫자를 담은 배열 arr이 입력되었을 때 이 수들의 최소공배수를 반환하는 함수, solution을 완성해 주세요.
제한사항
- arr은 길이 1이상, 15이하인 배열입니다.
- arr의 원소는 100 이하인 자연수입니다.
입출력 예
| arr | return |
|---|---|
| [2,6,8,14] | 168 |
| [1,2,3] | 6 |
나의 생각
모든 요소를 파악해야 된다고 생각했고, stack을 활용해서 2개의 요소의 최소공배수를 구하고,
다시 구한 최소공배수와 비교하는 식으로 해결해야겠다고 생각했다.
코드
- gcd와 lcm을 활용.
def gcd(a, b):
while b:
a, b = b, a % b
return a
def lcm(a, b):
return a * b // gcd(a, b)
def solution(arr):
answer = 0
while len(arr) > 1:
a = arr.pop(0)
b = arr.pop(0)
if a < b:
a, b = b, a
arr.append(lcm(a, b))
answer = arr[0]
return answer
- reduce()를 사용.
- gcd를 사용하는데
fractions.gcd() is deprecated. Use math.gcd() instead.이 문구가 나왔다. 이제 math.gcd()를 사용하자.
from functools import reduce
# from fractions import gcd
import math
def solution(arr):
return reduce(lambda a, b : a * b // math.gcd(a, b), arr)