Python/Coding Test

[Coding Test] N개의 최소공배수

gangee 2024. 3. 9. 23:34
728x90
반응형

프로그래머스 코딩테스트 2단계 : N개의 최소공배수

문제 설명

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

문제 풀이

  • math 라이브러리의 gcd 함수를 이용해 최대공약수를 구해줌
  • 최대공약수를 이용해 최소공배수 구하는 알고리즘 만듦
  • 반복문을 이용해 arr 배열의 마지막 값 2개를 지우면서 lcm을 구한 후 나온 값을 다시 추가
  • arr 배열의 길이가 1이 될 때까지 반복한 후 1이 되면 arr[0] 출력
  • pop함수에 대해 자세히 알고싶다면 아래 블로그 링크 참고
    [기초문법] Stack, Queue

정답 코드

import math
def lcm(x, y):
        return x*y // math.gcd(x, y)

def solution(arr):

    while True:
        arr.append(lcm(arr.pop(), arr.pop()))
        if len(arr) == 1:
            return arr[0]
  • 최소공배수를 구할 때는 2개의 숫자만 쓰이기 때문에 두개의 최소공배수를 구한 후 그 결과값을 다음 수와 비교해 한번 더 최소공배수를 구하는 방법이 문제의 핵심
* 이 문제는 프로그래머스 코딩테스트 연습 2단계 문제입니다.
728x90
반응형