[Go] 프로그래머스 연습문제 LV1 - 최대공약수와 최소공배수

문제 설명

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

제한 사항

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

입출력 예

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

입출력 예 설명

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

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

내코드

func solution(n int, m int) []int {
    arr := make([]int,2,2)
    if n<m {
         n, m = m, n   
    }
    for i:=1; i<=n; i++{
            if n%i+m%i==0{
                arr[0] = i
                arr[1] = n/i*m/i*i
            }
        }
    return arr
}

• 슬라이스 생성
• n과 m 비교해서 작은 수를 n으로 스위칭
• for 문으로 n이 될 때까지 증가 하면서 나머지 0인 숫자를 arr[0]에 저장.
• arr[1]에는 최소공배수가 들어감.
• 다른 사람 코드를 봐도 개인적으로 간단하다고 생각함.
• 다만 for loop에서 n부터 역으로 감소하며 return을 넣어서 하는 것도 좋다고 생각함(n이 최대공약수 일 시 연산 횟수 최소화).

func solution(n int, m int) []int {
    g := n * m
    for n != 0 {
        if n < m {
            n, m = m, n
        }
        n -= m
    }
    return []int{m, g / m}
}