프로그래머스 - 멀쩡한 사각형 (Python)

Summer/Winter Coding(2019)

코딩테스트 연습 - 멀쩡한 사각형 | 프로그래머스 (programmers.co.kr)

코딩테스트 연습 - 멀쩡한 사각형

 

초5 수학 경시대회를 풀 수 있는 수학적 능력이 요구된다,,,ㅎ

[초5경시] 대각선이 지나는 격자점의 개수 -청주 금천동(금천광장) 수학학원 ★오르비수학★ : 네이버 블로그 (naver.com)

[초5경시] 대각선이 지나는 격자점의 개수 -청주 금천동(금천광장) 수학학원 ★오르비수학★

 

Step1. 대각선이 지나는 격자점의 개수

Step2. 대각선이 지나는 사각형의 개수

 

import math

def solution(w,h):
    answer = w * h
    
    # 최대공약수
    gcd_ = math.gcd(w, h)
    
    # 격자점이 없는 경우
    if gcd_ == 1:
        answer -= w + h - 1
    # 격자점이 있는 경우
    else:
        answer -= w + h - gcd_
        
    return answer

 

위의 코드를 짧게 줄이면 아래와 같다.

import math

def solution(w,h):
    return w * h - (w + h - math.gcd(w, h))