[프로그래머스] 멀쩡한 사각형

프로그래머스 멀쩡한 사각형 - 문제 보러 가기

입력 조건

가로 길이가 Wcm, 세로 길이가 Hcm인 직사각형 종이가 있습니다. 종이에는 가로, 세로 방향과 평행하게 격자 형태로 선이 그어져 있으며, 모든 격자칸은 1cm x 1cm 크기입니다. 이 종이를 격자 선을 따라 1cm × 1cm의 정사각형으로 잘라 사용할 예정이었는데, 누군가가 이 종이를 대각선 꼭지점 2개를 잇는 방향으로 잘라 놓았습니다.
그러므로 현재 직사각형 종이는 크기가 같은 직각삼각형 2개로 나누어진 상태입니다. 새로운 종이를 구할 수 없는 상태이기 때문에, 이 종이에서 원래 종이의 가로, 세로 방향과 평행하게 1cm × 1cm로 잘라 사용할 수 있는 만큼만 사용하기로 하였습니다.
가로의 길이 W와 세로의 길이 H가 주어질 때, 사용할 수 있는 정사각형의 개수를 구하는 solution 함수를 완성해 주세요.

풀이

테스트 조건에서 주어진 조건은 8X12로 주어졌다.
그림을 잘 살펴보면 2X3만큼의 양만큼 지나가고있다.
8과 12의 최소 공배수는 4로 대각선이 지나가는 한칸의 사각형은 다음과 같다.
(8/4)+(12/4)-1 해당식을 전제로 코드를 작성해보았다.

코드

Go Version

func solution(w, h int) (answer int64) {
    gcd := 0
    if w < h {
        gcd = getGcd(w, h)
    } else {
        gcd = getGcd(h, w)
    }
    small := (w/gcd) + (h/gcd) - 1
    answer = (int64(w) * int64(h)) - int64(small * gcd)
    return
}

func getGcd(w, h int) (gcd int) {
    for idx := w; idx > 0; idx-- {
        if (w % idx == 0) && (h % idx == 0) {
            gcd = idx
            break
        }
    }
    return
}