[프로그래머스/Level1/파이썬3] 최대공약수와 최소공배수

코딩테스트 연습 - 최대공약수와 최소공배수

 

 

 

문제 설명

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

 

 

 

제한 사항

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

 

 

입출력 예

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

 

 

입출력 예 설명

- 입출력 예#1

위의 설명과 같습니다.

 

- 입출력 예#2

자연수 2와 5의 최대공약수는 1, 최소공배수는 10이므로 [1,10]을 리턴해야 합니다.

 

 

 

코드

def gcd(n,m):
    while m:
        n, m = m, n%m
    return n
 
def lcm(n,m):
    return n*m // gcd(n,m)
 
def solution(n,m):
    answer = [gcd(n,m), lcm(n,m)]
    return answer

 

1~4번째 줄 : 최대 공약수를 구하는 코드이다. while에 있는 m이 0이 아닐 때까지 반복되며 m이 0이 되는 경우는 n%m이 0 즉 나머지가 0이되면 m이 0이되어 while문을 멈추고 n을 return하게 된다.

 

6,7번째 줄 : 최소 공배수를 구하는 코드이다. 최소공배수를 구하는 방식에 따라 n과 m을 곱하고 이를 최대 공약수를 나눈 값의 몫을 return한다.

 

9~11번째 줄 : 위에서 구한 최대 공약수와 최소 공배수를 list로 만들어 answer에 값을 저장한 후, return 한다.

 

 

def solution(n, m):
    c, d = max(n, m), min(n, m)
    while d:
        c, d = d, c%d
    answer = [c, int(n*m/c)]
    return answer