[프로그래머스] 최대공약수 최소공배수

 

문제 설명

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

제한 사항

-행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

입출력 예

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

입출력 예 설명

입출력 예 #1

위의 설명과 같습니다.

입출력 예 #2

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

​

접근법

최대공약수 같은 경우에는 두 수의 공통된 약수들 중 가장 큰 수를 의미하므로, 모든 약수들을 리스트 형태로 받아 계산해 보겠습니다. 최소공배수 같은 경우는 입력받은 두 수의 곱을 최대공약수로 나눈것과 같으므로 최대공약수만 구한다면 쉽게 구할 수 있습니다.

​

나의 코드

def solution(n, m):
    n_list = [num for num in range(1,n+1) if n%num==0]
    m_list = [num for num in range(1,m+1) if m%num==0]
    max_list = list(set(n_list) & set(m_list))
    return [max(max_list),n*m/max(max_list)]

다른 코드

from fractions import gcd
def solution(n, m):
    return [gcd(n,m),n*m/gcd(n,m)]