두 수의 최소공배수(Least Common Multiple)란 입력된 두 수의 배수 중 공통이 되는 가장 작은 숫자를 의미합니다. 예를 들어 2와 7의 최소공배수는 14가 됩니다. 정의를 확장해서, n개의 수의 최소공배수는 n 개의 수들의 배수 중 공통이 되는 가장 작은 숫자가 됩니다. n개의 숫자를 담은 배열 arr이 입력되었을 때 이 수들의 최소공배수를 반환하는 함수, solution을 완성해 주세요.
arr | result |
---|---|
[2,6,8,14] | 168 |
[1,2,3] | 6 |
from math import gcd
def solution(arr):
answer = 0
stack = []
arr = sorted(arr)
stack.append(arr[0])
for i in range(1,len(arr)):
GCD = gcd(arr[i],stack[-1])
lcd = arr[i] * stack[-1] // GCD
stack.append(lcd)
answer = max(stack)
return answer
우선 최소공배수는 최대공약수 gcd(a,b)
함수를 활용해서 구할 수 있다. 이때 주의할 점은 a>b인것을 유념해야한다.
그리고 배열 arr
을 돌면서 순차적으로 최소공배수 값이 stack에 쌓이게 되고, 오름차순으로 값이 쌓이기 때문에 max
값을 통해 구하면 최소공배수를 return한다.
from math import gcd
def nlcm(num):
answer = num[0]
for n in num:
answer = n * answer / gcd(n, answer)
return answer
로직은 같은데, 내가 너무 어렵게 구현을 했구나,, 생각했다.
배열을 돌면서 누적하면서 곱해주면, 최종 최소공배수를 구할 수 있는 것은 똑같다.