본문 바로가기

Algorithm & SQL/Programmers

[Programmers] [Python] 최소공배수와 최대공약수

최대공약수와 최소공배수


문제 설명


두 수를 입력받아 두 수의 최대공약수와 최소공배수를 반환하는 함수, solution을 완성해 보세요. 배열의 맨 앞에 최대공약수, 그다음 최소공배수를 넣어 반환하면 됩니다. 예를 들어 두 수 3, 12의 최대공약수는 3, 최소공배수는 12이므로 solution(3, 12)는 [3, 12]를 반환해야 합니다.

제한 사항


  • 두 수는 1이상 1000000이하의 자연수입니다.

입출력 예


n m return
3 12 [3, 12]
2 5 [1, 10]

제출코드


def solution(n, m):
   for i in range(1, n+1):
       if n % i == 0:
           if m % i == 0:
               gcd = i
   if gcd == 1:
       lcm = n*m
   else:
       lcm = gcd *(n // gcd) *(m // gcd)
   return [gcd, lcm]

코드설명


Line 2~5 : 1~n 까지의 수를 순회하며 nm을 동시에 나눌 수 있는 수 중 가장 큰 값, 즉 최대공약수를 구한다.

Line 6~7 : 만일, 최대공약수가 1인 경우, 최소공배수 = n * m으로 구해준다.

Line 8~9 : 그외의 경우에는 일반적으로 최소공배수를 구하는 공식을 이용해 구해준다.