[프로그래머스] Lv.2 N개의 최소공배수 (Python / Java)

 

 

코딩테스트 연습 - N개의 최소공배수 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

문제 설명

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

제한 사항

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

입출력 예

---

문제 풀이

이 문제는 A, B, C 세 수의 최소공배수를 구한다고 범위를 좁혀 생각을 하고 문제를 풀었다.
A와 B의 최소공배수를 먼저 구하고, 그 결과와 C의 최소공배수를 다시 구하는 방식으로 세 수의 최소공배수를 구하기 때문에 세 수 이상의 최소공배수는 LCM(A,B,C) = LCM(LCM(A,B),C) 방식으로 문제를 풀었다. 

[Python]

import math
from functools import reduce

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

def solution(arr):
    return reduce(lcm, arr)

 

[Java]

import java.util.Arrays;

class Solution {
    public int solution(int[] arr) {
        return Arrays.stream(arr).reduce(1, (a,b) -> lcm(a, b));
    }
    
    public static int gcd (int a, int b) {
        while (b > 0) {
            int tmp = b;
            b = a % b;
            a = tmp;
        }
        return a;
    }
    
    public static int lcm(int a, int b) {
        return a * b / gcd(a,b);
    }
}