[ Programmers / CodingTest / Python ] 최대공약수와 최소공배수

문제 설명

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

접근 방법

두 수에 대해서 나눌 수 있는 수로 전부 나눈다. 이때 나누는 수를 임시 배열에 저장한다. 최대 공약수는 여기서 임시 배열에 있는 수들을 모두 곱하면 얻을 수 있고, 최소 공배수는 이 임시 배열의 수들의 곱과 나눠지고 난 수들의 곱을 곱하면 구할 수 있다.

최소 공배수가 1인 경우에는 나눌수 있는 수가 없기 때문에 임시 배열이 빈 배열로 남아있게 되므로 임시 배열에 어떤 수를 곱해도 영향을 안주는 1을 넣어둔다.

  • 나누는 수를 담아 둘 임시 배열 tmp에 1을 넣어 선언한다.
  • 나눌 수 있는 수를 모두 구하기 위해 사용할 변수 cur을 1로 선언한다.
  • n이 1보다 크고 m이 1보다 크고 cur이 n, m중 더 작은 수보다 작거나 같을 동안 반복하는 while문을 돌린다.
    -> cur을 1 증가시킨다.
    -> 만약 n, m이 모두 cur로 나눠 떨어진다면,
    --> tmp에 cur을 넣는다.
    --> n, m을 cur로 나눈 값으로 갱신한다.
    --> cur을 1로 갱신한다.
    -> 나눠 떨어지지 않는다면 다음 반복을 진행한다.
  • 최대 공약수를 저장할 변수 gcd와 최소 공배수를 저장할 변수 lcm을 1로 선언한다.
  • tmp를 순회하는 i에 대한 for문을 돌린다.
    -> gcd에 i를 곱한다.
  • lcm에 n*m*gcd를 저장한다.
  • answer를 [gcd, lcm]으로 저장한다.
  • answer를 반환한다.

solution.py

def solution(n, m):
    tmp=[1]
    cur=1
    while n>1 and m>1 and cur<=min(n, m):
        cur+=1
        if n%cur==0 and m%cur==0:
            tmp.append(cur)
            n, m=n//cur, m//cur
            cur=1
        else:
            continue
    gcd, lcm=1, 1
    for i in tmp:
        gcd*=i
    lcm=n*m*gcd
    answer=[gcd, lcm]
    return answer

좋은 웹페이지 즐겨찾기