[Python] Lv1.최대공약수와 최소공배수

https://programmers.co.kr/learn/courses/30/lessons/12940

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

 

 

 

 

 

문제 설명

두 수를 입력받아 두 수의 최대공약수와 최소공배수를 반환하는 함수, 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]을 리턴해야 합니다.

 

나의 접근방법 및 풀이

for() 문하고 if() 문을 어떻게 써야 하나 생각하고 있었는데, 예전에 배웠던 유클리드 호제법이 생각났다.

우선 최소공배수를 구할 때 사용하기 위해서, 원본 n, m값을 따로 변수에 저장한다.

그리고 대소 비교를 해서 n에는 작은 값, m에는 큰 값을 넣는다.

m을 n으로 나눈 나머지 r을 생성하고, m자리에는 n을, n자리에는 r을 넣어 r 즉, 나머지가 0이 될 때까지 반복한다.

r이 0이 되었을 때의 n이 최소 공약수이다.

최대 공약수는 기존의 매개변수 m*n을 최소 공약수로 나눈 것이므로, a*b/n이다.

def solution(n, m):
    a,b = n, m
    if n > m:
        n, m = m, n     ## n이 최소, m이 최대
    while(m%n):     ## 나머지가 0이 되면 벗어남
        r = m%n
        m = n
        n = r
        
    return [n, m*n/n]