[Programmers] PS - 003

문제 출처 : 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]

풀이

def solution(n, m):
    a = []
    b = []
    c = []
    ans = []
    mm = min(n, m)
    xx = max(n, m)

    for i in range(1, mm+1):
        if mm % i == 0:
            a.append(i)
    for i in range(1, xx+1):
        if xx % i == 0:
            b.append(i)
    for i in a:
        for j in b:
            if i == j:
                c.append(j)
    ans.append(max(c))
    
    if (mm in a) and (mm in b):
        ans.append(max(b))
    else:
        if len(c) == 1:
            ans.append(max(a) * max(b))
        else:
            ans.append((max(a) * max(b))/max(c))
    return ans