[Python] 프로그래머스 (최대공약수와 최소공배수)

문제 설명

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

제한 사항

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

입출력 예

nmreturn

3	12	[3, 12]
2	5	[1, 10]

입출력 예 설명

입출력 예 #1
위의 설명과 같습니다.

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

---

문제풀이

• n과 m을 인수분해 한다.
• n, m의 최대공약수(GCD)를 구한다. (공통된 약수중 가장 큰 수)
• 최소공배수(LCM) =  n*m / GCD이다.

최대공약수와 최소공배수의 관계 참고: https://calcproject.tistory.com/770

n = 3
m = 12

def solution(n, m):
    answer = []
    n_list = []
    m_list = []
    for i in range(1, n+1):
        if n % i == 0:
            n_list.append(i)
    
    for i in range(1, m+1):
        if m % i == 0:
            m_list.append(i)

    LCM = list(set(n_list) & set(m_list))
    LCM = sorted(LCM)
    GCD = n*m // LCM[-1]
    answer.append(LCM[-1])
    answer.append(GCD)
    return answer

print(solution(n,m))