멀쩡한 사각형

링크 : https://programmers.co.kr/learn/courses/30/lessons/62048#

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

(프로그래머스>코딩테스트 연습>Summer/winter Coding(2019)>멀쩡한 사각형)

 

---

 

<문제>

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

• W, H : 1억 이하의 자연수

 

---

 

<내가 작성한 코드>

def solution(w,h):
    answer = 0
    if w<=1 or h<=1:
        return 0
    elif w==h:
        return w*h-w
    else:
        for i in range(1, w):
            answer+=(h*i//w)
        return answer*2

일일히 계산하지 않아도 되는 경우는 예외로 처리해주는게 시간 단축에 효율적이다.

 

 

 

 

<다른 사람 코드를 참고하여 작성한 코드>

from math import gcd
def solution(w,h):
    return w*h-(w/gcd(w,h)+h/gcd(w,h)-1)*gcd(w,h)

참고 : https://programmers.co.kr/learn/courses/30/lessons/62048/solution_groups?language=python3

---

 

<참고한 문법>

gcd() 함수 : 유클리드 호제법을 이용하여 두 수의 최대공약수를 구하는 함수

def gcd(x,y):
	while y:
		x,y = y,x%y
    return x