본문 바로가기

Algorithm

[프로그래머스] N개의 최소공배수(Python)

반응형

https://programmers.co.kr/learn/courses/30/lessons/12953

 

코딩테스트 연습 - N개의 최소공배수

두 수의 최소공배수(Least Common Multiple)란 입력된 두 수의 배수 중 공통이 되는 가장 작은 숫자를 의미합니다. 예를 들어 2와 7의 최소공배수는 14가 됩니다. 정의를 확장해서, n개의 수의 최소공배

programmers.co.kr

문제 설명

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

제한 사항

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

입출력 예

arr result
[2,6,8,14] 168
[1,2,3] 6

풀이 설명

1. 최소공배수를 구하는 공식인 "a * b / a와 b의 최대공약수" 를 활용한다.

2. 파이썬의 최대공약수 메서드인 gcd를 이용한다.

3. arr의 원소가 하나만 남을때까지

  3-1. 원소 2개를 꺼내어 최소공배수를 구한 후 arr에 삽입한다.

4. 하나만 남은 원소를 리턴한다.

Python3 코드

from math import gcd

def solution(arr):
    while True:
        a, b = arr.pop(), arr.pop()
        arr.append(a * b // gcd(a, b))
        if len(arr) == 1:
            return arr[0]
반응형