Python/Coding Test
[Coding Test] N개의 최소공배수
gangee
2024. 3. 9. 23:34
728x90
반응형
문제 설명
두 수의 최소공배수(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
반응형