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

dobyming·2023년 1월 20일
0

문제 설명

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

제한 사항

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

입출력 예

arrresult
[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

로직은 같은데, 내가 너무 어렵게 구현을 했구나,, 생각했다.
배열을 돌면서 누적하면서 곱해주면, 최종 최소공배수를 구할 수 있는 것은 똑같다.

0개의 댓글